ModuleNotFoundError: No module named 'torch'和torch.cuda.is_available返回False我悟了!!!
*大坑:cuda、torch、torchvision的版本匹配问题!!!*
报错显示
cmd、pycharm中出现torch.cuda.is_available()返回False
ModuleNotFoundError: No module named 'torch'
可能的原因及解决办法:
原因一:在当前使用的虚拟环境中未安装torch
1查看cuda
打开终端,先查看自己的CUDA版本(决定了后续要下载torch版本)
nvidia-smi
2下载torch、torchaudio、torchvision
需要下载三个文件:torch、torchaudio、torchvision(后缀都是.whl)
我的cuda版本是10.2,在torch的官网下载网页中找到对应的版本(可以上网搜一下版本对应)
注意:版本下错了也会是false!
我下载的就是这三个文件
解释一下子
下载方式
- 网页下载(太慢了,不推荐)
- 迅雷下载(速度很快,感觉没用五分钟)
点击新建,复制粘贴要下载的文件的链接,记住下载的路径(后续pip的时候要用)
3pip安装
打开cmd,进入之前下载文件的路径下(或者在存放三个文件的目录下,右击打开终端)
pip install torch-1.7.1+cu101-cp38-cp38-win_amd64.whl
pip install torchvision-0.8.2+cu101-cp38-cp38-win_amd64.whl
pip install torchaudio-0.7.2-cp38-none-win_amd64.whl
4测试
进入python模式
python
import torch
torch.cuda.is_available()
返回结果为True,表示安装成功!
也可以在pycharm和jupyter中测试
原因二:已经安装过但仍然返回False
1检查是否为GPU
- 查看自己到底安装的版本是CPU**还是GPU*!
- 查看自己安装的版本(cuda、torch、torchvision的版本匹配问题,参考原因一)
进入python模式
python
import torch
print(torch.__version__)
我之前从镜像网上下载,安装的就是cpu,结果报的就是false,后来卸载,重新安装Gpu后,成功了!
2卸载重下
如果下载错了,快删掉!
pip uninstall torch...
pip uninstall torchvision...
pip uninstall torchaudio...
然后去官网网址上下载pyTorch的GPU版!
.
.
…后续请参考上述原因一
从这点上去
…
菜鸟编辑,欢迎指出错误~