解决print(torch.cuda.is_available())返回false的问题

这篇具有很好参考价值的文章主要介绍了解决print(torch.cuda.is_available())返回false的问题。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

  1. 问题简述

笔者在安装了CUDA、cuDNN、Torch后,发现仍无法调用GPU;

print(torch.cuda.is_available())

始终返回false,查阅了网上各种方法都无效;

解决print(torch.cuda.is_available())返回false的问题

NVIDIA控制面板

nvidia-smi
解决print(torch.cuda.is_available())返回false的问题

CUDA支持版本

检查了CUDA版本、cuDNN版本也都正确;

  1. 解决方案

直到一次偶然,我在Pycharm中检查torch版本时,

import torch
print(torch.__version__)

返回的居然是:

x.x.x+cpu

遂恍然大悟,我原来装了个CPU版本的torch!这里提醒小伙伴,不要直接在pycharm里的PyPI里装torch,这里装的只是CPU版本。

读者可以复制这段代码自查:print(torch.__version__)。如果返回也是CPU版本,那么笔者建议按照以下步骤正确装入GPU版本torch:

  1. 进入官网

选择对应的版本pip指令

解决print(torch.cuda.is_available())返回false的问题
  1. 复制pip指令安装torch

解决print(torch.cuda.is_available())返回false的问题
  1. 验证

import torch
from torch.backends import cudnn

print(torch.__version__)
print(torch.cuda.device_count())
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())

结果:

解决print(torch.cuda.is_available())返回false的问题

返回True ~文章来源地址https://www.toymoban.com/news/detail-410736.html

到了这里,关于解决print(torch.cuda.is_available())返回false的问题的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包