一、运行代码时出现下面错误:
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.从阿里镜像站下载镜像
进入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报错问题。
四、总结
查看CUDA版本 | 1.输入指令 nvidia-smi 2.CUDA 11.6 |
阿里云镜像文件 | pytorch-wheels安装包下载_开源镜像站-阿里云 (aliyun.com) 找CU116中的torch和torchvision 参考Torch 、torchvision 、Python 版本对应关系以及安装 GPU 或 CPU 版本的 pytorch_torch torchvision_xzw96的博客-CSDN博客 确定最终的torch和torchvision版本,pip本地镜像安装。 |