(原创)torch.cuda.is_available()返回false的解决方法

耗时数天,今天终于解决了

狂喜!!!

如果你目标是在虚拟环境中安装Pytorch,

如果你确定你的CUDA版本和CUDNN版本、torch版本、torchaudio版本、torchvision版本对应,

但当你在虚拟环境中ipython然后import torch然后torch.cuda.is_available()后仍然返回false,

那么请检查一下在虚拟环境外你是否有安装torch,检查下那个torch是否是cpu版本,因为我发现,我在虚拟环境中运行ipython然后import torch后,它好像import的是虚拟环境外的torch,而我虚拟环境外的刚好是cpu版本,所以无论怎么变虚拟环境内的pytorch它都返回false

因此,我先删除虚拟环境外的Pytorch的包,然后参考了以下链接的方法:torch.cuda.is_available()返回false——解决办法_cuda available false_Nefu_lyh的博客-CSDN博客jicon-default.png?t=N7T8https://blog.csdn.net/qq_46126258/article/details/112708781

把在虚拟环境里Pytorch的几个包复制到了虚拟环境外,结果再在虚拟环境中ipython然后import torch然后torch.cuda.is_available()后,显示True,结果如下图:

但是,我后来发现ipython后环境好像换到了base环境下,所以我在虚拟环境中输入python后再import torch然后torch.cuda.is_available(),最后返回依然为True

不过其间也遇到了一些小bug:

第一个是ModuleNotFoundError: No module named 'typing_extensions',这个直接conda install typing_extensions就好。

还有一个numpy方面的报错,我直接conda install numpy就好了。

另外,附上我安装的包的版本:

我的驱动版本:

安装CUDA、CUDNN、Pytorch的过程中也有很多其他可能踩坑的点,比如以下几点:

1.安装pytorch的命令应该pip而不是conda(这个图就浅浅借一下链接作者的吧~~)

2.pip时报错:首先在命令行操作时不能挂梯子,其次还报错的话应该先把三个whl包下载到本地再pip

pip install torch-2.0.1+cu118-cp311-cp311-win_amd64.whl 
pip install torchaudio-2.0.2+cu118-cp311-cp311-win_amd64.whl 
pip install torchvision-0.15.2+cu118-cp311-cp311-win_amd64.whl

以上一些点其他作者也写过很多,这里就不过多赘述了

最后祝大家配环境的过程中少踩坑,少遇bug!!!

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

一溪风月_

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

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

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

打赏作者

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

抵扣说明:

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

余额充值