安装顺序如下:
cuda(11.4)-->cudatoolkit(11.3)-->pytorch(1.10.0),pytorchvision(0.11.1)-->mmcv-full(1.4.2)

  1. 创建python3.7的虚拟环境miaod并激活
conda create -n miaod python=3.7 -y
conda activate miaod
  1. 安装pytorch ,torchvision和cudatoolkit
    不确定版本的话看这个https://pytorch.org/
conda install pytorch cudatoolkit=11.3 torchvision -c pytorch

查看安装成功
 

image.png


但是因为conda下载太慢,经常会断,而且我这里遇到了点问题,不知道是不是镜像源配置问题(解决:当时把整个虚拟环境删掉,重新建了),所以可以采用wget方式安装,这里不赘述.

  1. 安装mmcv-full
    这里我执行了
pip install mmcv-full==1.4.2 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html

官方给出的命令如下:

pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/{cuda_version}/{torch_version}/index.html

其中,{ }中的版本需要替换成对应版本,不知道的可以查看https://github.com/open-mmlab...
 

image.png


点击标记的地方,打开里面的网址,可以查看更多的mmcv版本

image.png

  1. 查看是否安装成功
    pip list

    image.png

我的神奇出错步骤

因为要cuda和torch版本要对应,所以参考链接

但是我的cuda是11.4,所以自作聪明把cu114改成了cu114,造孽呀,试了好多次才发现错在一个版本数字上。
个人输入:

pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu114/torch1.10.0/index.html

更多推荐