当前位置: 首页 > >

window10+pytorch+cuda+torch.cuda.is_available()=False 驱动器版本问题

发布时间:

1.安装pytorch:


网址:https://pytorch.org/


我是用pip去安装的,我的python版本是3.7,注意在cuda那里一定不能选none,我直接选的是10.1版本的pytorch.cuda,将下面的那段command运行一下



中途你会发现安装的速度很慢,那么你就直接到:https://download.pytorch.org/whl/torch_stable.html这个地址下载相关的轮子,具体的轮子幸好会出现在你一开始用那段cmd命令下,下载后直接使用


pip install xxx.whl

?


2.测试pytorch和它的cuda的安装版本和是否能够调用:


import torch
print(torch.__version__ )
print(torch.version.cuda)
print(torch.cuda.is_available())

此时显式出来的结果是:


1.3.1


10.1


False


说明torch可以使用,但是最后的False说明使用cuda不成功


3.在实际调用时,说我的cuda版本不匹配,因此我考虑是不是我本身的驱动器问题。打开看一下驱动器的版本,打开方式为:http://baijiahao.baidu.com/s?id=1601797854538600303&wfr=spider&for=pc


windows系统查看NVIDIA显卡支持的CUDA版本,网址:https://blog.csdn.net/LearnToPain/article/details/100902239


如果不匹配,那么说明可能是驱动器版本太低


4.重新下载驱动器,网址:https://www.nvidia.com/Download/index.aspx?lang=en-us



5.安装驱动器


6.重新测试一下:


import torch
print(torch.__version__ )
print(torch.version.cuda)
print(torch.cuda.is_available())

结果:



?


?



友情链接: