デムックスのビデオメモリ不足に対する2つの解決策
Demucsを使用してハイレゾリューション・オーディオを処理する場合、グラフィックスカードのビデオメモリ不足の問題に遭遇することがあります。ここでは2つの解決策をご紹介します:
オプション1:処理パラメータの最適化
- 利用する
--segment 8
パラメーターを使用すると、音声を8秒に分割して処理することができます。 - 増加
--two-stems
パラメーターは特定のトラックのみ(例えばボーカルのみ)を抽出し、計算量を減らす。 - 利用する
-j 4
GPUの圧力を共有するためのパラメータ対応マルチコアCPU補助計算
オプション2:コンピューティング・デバイスの切り替え
- 強制CPU計算:追加
-d cpu
パラメトリック - 環境変数の設定
PYTORCH_NO_CUDA_MEMORY_CACHING=1
CUDAメモリーキャッシュを無効にする - Macユーザーの場合は
-d mps
Apple Metalアクセラレーションを有効にする
補足:WAVファイルのサンプルレートを44.1kHzに下げることでも、メモリ消費量を大幅に削減できます。ビデオメモリが3GB以下の場合は、CPUソリューションを優先することをお勧めします。
この答えは記事から得たものである。Demucs: 音楽トラックを分離するための無料のオープンソースツールについて