根据电脑平台比如你的电脑是Windows或者是Linux,虚拟环境里面的Pytorch版本和CUDA版本选择最合适的MMCV版本安装
①点击这里访问👉:电脑平台&CUDA版本&Torch版本一一对应关系网址
② 点击这里访问👉:安装CUDA Toolkit Archive 网址
我这里默认选择11.7的版本,接着就是按照指引选择平台进行安装,如果是Linux系统,选择Run文件,如果是WIndows选择exe local方式
关于如何查询CUDA版本,在命令行输入
nvcc -V
③安装指定版本的虚拟环境里面的cuda包
PS:②和③安装的cuda版本要一模一样!
②的我一般安装了就是11.7版本
③的就是根据CUDA Toolkit Archive的11.7的版本去搜索对应虚拟环境里面的cuda对应的torch包