在你的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
复制红线标记的网址,打开后显示:
找到自己对应的版本,下载之前先打开cmd终端执行下面一句话查看自己下载的python支持的文件名:
pip debug --verbose
我就选择红线标记的这个版本找到对应链接下载好了
将torch和torchvision的.whl都下载好,下载完成后,打开cmd,注意一定要用cmd!!!
进入到下载文件的目录中,打开自己的虚拟环境,例如我把他放在了D盘
测试一下:
python
import torch
torch.cuda.is_available()
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/
转载文章受原作者版权保护。转载请注明原作者出处!