解决Demucs显存不足的两种方案
当使用Demucs处理高分辨率音频时,可能遇到显卡显存不足的问题。以下是两种解决方案:
方案一:优化处理参数
- 使用
--segment 8
参数降低内存需求,该参数将音频分割成8秒的片段处理 - 添加
--two-stems
参数只提取特定音轨(如仅提取人声),减少计算量 - 使用
-j 4
参数启用多核CPU辅助计算,分担GPU压力
方案二:切换计算设备
- 强制使用CPU计算:添加
-d cpu
参数 - 设置环境变量
PYTORCH_NO_CUDA_MEMORY_CACHING=1
禁用CUDA内存缓存 - 对于Mac用户,可使用
-d mps
启用Apple Metal加速
补充建议:将WAV文件采样率降至44.1kHz也能显著降低内存消耗。显存低于3GB时建议优先考虑CPU方案。
本答案来源于文章《Demucs:免费分离音乐音轨的开源工具》