无法调用自己电脑的gpu,即torch.cuda.is_available()返回false

在你的CUDA,cuDNN, torch 版本对应的情况下检查torch版本

python
import torch
print(torch.__version__)

居然是+cpu,明明自己下载的是gpu版本

1.9.1+cpu

搜寻了一圈从该博主下找到了答案如图:(23条消息) torch.cuda.is_available()返回false——解决办法_Nefu_lyh的博客-CSDN博客_torch.cuda.is_available

无法调用自己电脑的gpu,即torch.cuda.is_available()返回false

无法调用自己电脑的gpu,即torch.cuda.is_available()返回false

接下来去官网下载gpu版本的torch

进入官网Start Locally | PyTorch

无法调用自己电脑的gpu,即torch.cuda.is_available()返回false

复制红线标记的网址,打开后显示:

无法调用自己电脑的gpu,即torch.cuda.is_available()返回false

找到自己对应的版本,下载之前先打开cmd终端执行下面一句话查看自己下载的python支持的文件名:

pip debug --verbose

无法调用自己电脑的gpu,即torch.cuda.is_available()返回false

我就选择红线标记的这个版本找到对应链接下载好了

将torch和torchvision的.whl都下载好,下载完成后,打开cmd,注意一定要用cmd!!!

进入到下载文件的目录中,打开自己的虚拟环境,例如我把他放在了D盘

无法调用自己电脑的gpu,即torch.cuda.is_available()返回false

无法调用自己电脑的gpu,即torch.cuda.is_available()返回false

测试一下:

python
import torch
torch.cuda.is_available()

无法调用自己电脑的gpu,即torch.cuda.is_available()返回false

Original: https://blog.csdn.net/qq_36521174/article/details/123654700
Author: cc和ss
Title: 无法调用自己电脑的gpu,即torch.cuda.is_available()返回false

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/628017/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球