安装MMDetection:
中国移动九天·毕昇平台内置的pytorch1.8环境的pytorch,cuda版本为
import torch,torchvision
print(torch.__version__,torch.cuda.is_available())
!nvidia-smi
pytorch:1.8.0,CUDA Version:11.0
问题描述
使用pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.8.0/index.html
安装mmcv,会出现以下报错:Failed to build mmcv-ful
原因分析:
查看whl对应版本
可以发现没有对应于pytorch1.8与cuda11.0的版本
解决方案:
可以更改命令为pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.0/index.html
有问题欢迎+:xzming0908 讨论