解决torch、torchvision不匹配Torch not compiled with CUDA enabled问题

 一、运行代码时出现下面错误：

ModuleNotFoundError: No module named 'torch'

ModuleNotFoundError: No module named 'torchvision'

按照提示安装了torch和torchvision之后

 报错：

AssertionError: Torch not compiled with CUDA enabled

二、原因

pip自动安转的torch和torchvison的版本和CUDA版本不匹配。

三、解决办法

1.查看显卡驱动及CUDA版本

win+R调用，输入cmd

输入指令nvidia-smi

查看自己电脑的CUDA版本，我的对应的CUDA版本是11.6

2.从阿里镜像站下载镜像

进入阿里巴巴开源镜像站-OPSX镜像站-阿里云开发者社区

进入pytorch-wheels安装包下载_开源镜像站-阿里云

 找到自己需要的版本，其中cu116匹配CUDA11.6，以此类推找到符合自己的版本。

​

其中，cp37匹配虚拟环境中的Python版本，我的环境是Python3.7，下图框选的torch版本都可选。torchvision镜像下载重复这个操作。

​

这里要注意torch和torchvision的版本要对应。

参考：

PyTorch中torch、torchvision、torchaudio版本对应关系_torch与torchvision版本_Liekkas Kono的博客-CSDN博客

最终下载版本如下：

3.安装镜像文件

pip install +镜像文件绝对路径

解决AssertionError: Torch not compiled with CUDA enabled报错问题。

四、总结