MMCV 环境配置踩坑记

AudioSeal 音频水印系统

**AudioSeal** 是 Meta 开源的语音水印系统,用于 AI 生成音频的检测和溯源。

1.安装mmcv-full编译版本

MMCV 编译的版本安装会检测CUDA环境中的NVCC版本,然而conda环境下的cudatoolkit不包含nvcc,所以会MMCV编译会默认读取系统的CUDA环境而非conda的CUDA环境(由cudatoolkit创建)

conda 配置CUDA环境用cuda-toolkit而非cudatoolkit,前者包含了nvcc,是mmcv用来检查CUDA版本用的,同时设置CUDA_HOME在虚拟环境的根目录下,或者安装cudatoolkit后再单独安装nvcc

2.安装mmcv-full预编译好的版本

使用如下指令安装

pip --no-cache-dir install mmcv-full==1.5.0 -f https://download.openmmlab.com/mmcv/dist/cu112/torch1.10.1/index.html

将cu112 torch1.10.1替换为和当前cuda环境匹配的数字,torch 要和编译mmcv时的版本相同

同时-conda forge channel的torch 可能缺少个别mmcv运行时需要的.so文件,需要在-pytorch channel中安装torch

您可能感兴趣的与本文相关的镜像

AudioSeal 音频水印系统

AudioSeal 音频水印系统

语音合成
PyTorch
Cuda

**AudioSeal** 是 Meta 开源的语音水印系统,用于 AI 生成音频的检测和溯源。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值