关于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'
这是pycharm下的错误


可能的原因及解决办法:

原因一:在当前使用的虚拟环境中未安装torch

1查看cuda

打开终端,先查看自己的CUDA版本(决定了后续要下载torch版本)

nvidia-smi

cuda版本

2下载torch、torchaudio、torchvision

需要下载三个文件:torch、torchaudio、torchvision(后缀都是.whl)

我的cuda版本是10.2,在torch的官网下载网页中找到对应的版本(可以上网搜一下版本对应)

注意:版本下错了也会是false!
我下载的就是这三个文件whl文件下载
解释一下子
文件名解释

下载方式

  • 网页下载(太慢了,不推荐)
  • 迅雷下载(速度很快,感觉没用五分钟)
    点击新建,复制粘贴要下载的文件的链接,记住下载的路径(后续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中测试
pycharm测试


原因二:已经安装过但仍然返回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版!
.
.
…后续请参考上述原因一



菜鸟编辑,欢迎指出错误~

  • 2
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

@这个编程我不会

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值