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
2万+

被折叠的 条评论
为什么被折叠?



