torch.cuda.is_available() 的返回值为何一直是False?

初学者在使用PyTorch时遇到torch.cuda.is_available()返回False的问题,经过排查发现是安装的PyTorch版本不支持CUDA。通过更换为带GPU版本的PyTorch并重新安装后,问题得到解决,torch.cuda.is_available()返回True。
摘要由CSDN通过智能技术生成

为了复现一篇用神经网络处理图像的论文,第一次接触到Pytorch,于是就随手百度了一篇安装教程https://blog.csdn.net/weixin_43012220/article/details/83786637,万万没想到,这就是噩梦的开始……

运行代码的时候发现了非常严重的两个问题:1. 运行时间长 2.输出图像效果极差

根据以往使用waifu2x的经验,我立刻明白了这是大概是因为代码处理图像时没有用GPU而只用了CPU。重新运行代码,同时打开任务管理器的“性能”来查看设备使用情况,果不其然CPU几乎满载,GPU根本没有用上。

https://blog.csdn.net/nima1994/article/details/83001910,通过这个篇博客所讲内容,在代码中添加相关代码,所得结果全部都为无法使用GPU的表现。比如,将torch.cuda.is_available() 的返回值print出来,一直是False。

在网上查了很久很久,什么要额外再安装nividia cudnn之类的,试过了基本都没用,整个人完全陷入自闭状态……

后来又翻到了一两篇博客

https://blog.csdn.net/qq_38410428/article/details/82965499

  • 13
    点赞
  • 28
    收藏
    觉得还不错? 一键收藏
  • 18
    评论
评论 18
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值