torch.cuda.is_available()返回false

举报
风吹稻花香 发表于 2021/06/06 00:00:11 2021/06/06
【摘要】   torch.cuda.is_available()返回false     解决方法:已经安装torch的: import platform import torch sysstr = platform.system()use_cuda = torch.cuda.is_available()print('cuda',torch.cud...

 

torch.cuda.is_available()返回false

 

 

解决方法:已经安装torch的:


  
  1. import platform
  2. import torch
  3. sysstr = platform.system()
  4. use_cuda = torch.cuda.is_available()
  5. print('cuda',torch.cuda.is_available(),torch._C._cuda_getDriverVersion())
  6. daa=torch.Tensor([1,23])
  7. print(daa.cuda())

cuda10 不行的10000 411.31 升级完412.36

后来也不行,重装torch1.2.0

http://download.pytorch.org/whl/cu100/torch-1.2.0-cp36-cp36m-win_amd64.whl

cuda10 可以的 10020 442.19

没有安装torch的:

检查CUDA版本,目前Pytorch仅支持到CUDA 10.0版本。

                    检查驱动,更新到新版本。

 

cuda,cudnn安装后,我们可以验证是否配置成功,主要使用CUDA内置的deviceQuery.exe 和 bandwithTest.exe:
首先win+R启动cmd,cd到安装目录下的 ...\extras\demo_suite,然后分别执行bandwidthTest.exedeviceQuery.exe,应该得到下图:

 

Result=PASS,就验证通过了

 

 

文章来源: blog.csdn.net,作者:网奇,版权归原作者所有,如需转载,请联系作者。

原文链接:blog.csdn.net/jacke121/article/details/105289964

【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。