torch.cuda.is_available()返回False解决方案

在配置完环境以后运行程序,程序正常运行但是GPU没有任务,此时判断程序没有在GPU上运行。调试步骤:

  • 先查看cuda安装有没有问题:nvcc -V

                                

可以看到cuda驱动等安装没有问题。

  • 进程序去看.cuda()或者.todevice之类代码,看程序确实放在GPU上了,故排除

但是在查看代码是看到这里是一个and,参数args.no_cuda是设置的Flase,所以问题因为在torch.cuda.is_available(),没有获得cuda加速。

  • 查看深度学习环境搭建torch.cuda.is_available()这里返回了False,所以原因在这里

Torch官网的版本只提供了CUDA 9.2和CUDA 10.1的版本,我的cuda是10.0的。所以这里版本不对应导致Torch.cuda加速无法运行。

但是出现这个错误的原因有很多,但是因为环境配置的比较仓促,所以查看了一下环境配置,怀疑是pytorch安装版本不对应。

所以把torch,torchvision等相关安装包全部删除,安装适合版本的torch。

pip install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp37-cp37m-linux_x86_64.whl
pip install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp37-cp37m-linux_x86_64.whl

再次运行:

                                          

再次运行程序,程序就可以在GPU上运行啦。

 

  • 22
    点赞
  • 65
    收藏
    觉得还不错? 一键收藏
  • 6
    评论
根据引用和引用的内容,当调用torch.cuda.is_available()函数时,返回值为False表示没有可用的CUDA加速。这可能是因为您的深度学习环境没有正确安装CUDA或没有正确配置。要解决这个问题,您可以按照以下步骤进行操作: 1. 确保您的计算机上已正确安装了CUDA。您可以从NVIDIA官方网站上下载并安装适合您计算机配置的CUDA版本。 2. 如果您已经安装了CUDA,但仍然返回False,您需要检查您的深度学习框架是否与CUDA兼容。例如,在PyTorch中,您可以查看您安装的PyTorch版本是否与您的CUDA版本匹配。如果不匹配,您需要更新或重新安装PyTorch。 3. 另外,您还需要确保在运行代码或脚本时,没有禁用CUDA加速。根据引用的描述,参数args.no_cuda设置为False,表示CUDA加速是启用的。如果您的代码中存在类似的参数或设置,请确保将其设置为True或删除它,以便CUDA加速能够正常使用。 如果您已经按照上述步骤操作,并且仍然返回False,则可能是因为您的计算机硬件不支持CUDA。请确保您的计算机上有兼容CUDA的显卡。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *3* [Pytorch安装教程 及 解决 torch.cuda.is_available() 返回 False 的问题](https://blog.csdn.net/qq_43280818/article/details/104240772)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] - *2* [torch.cuda.is_available()返回False解决方案](https://download.csdn.net/download/weixin_38723699/13749367)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值