mmcv安装/更新过程记录

爱吃棒棒坤的GGBond

已于 2024-05-04 15:14:06 修改

阅读量1.5k

点赞数

文章标签： pytorch 深度学习 python

于 2022-06-14 15:38:27 首次发布

本文链接：https://blog.csdn.net/m0_46690805/article/details/125279141

版权

1.需要了解自己电脑的一些配置情况：

CUDA版本和 pytorch版本

2.获取CUDA版本方法/ pytorch版本的方法(通过以下代码获取cuda版本和pytorch版本)：

import torch
torch.version.cuda  #获取cuda版本。 不要通过 nvidia-smi 这种方式获取cuda版本！！！
torch.__version__   #获取pytorch版本

3.执行以下代码(安装 — mmcv 2.2.0 文档)：

pip install mmcv-full==1.3.9 -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.7.0/index.html 
##其中1.3.9是我们要安装的mmcv的版本号，根据需求自己选择对应版本号即可；
##cu111是我们的CUDA版本即11.1；torch1.7.0是我们的pytorch版本号。