首页 > 其他 > 详细

安装pytorch时,torch.cuda.is_available()返回false的解决办法

时间:2021-07-17 11:00:45      阅读:17      评论:0      收藏:0      [点我收藏+]

????最近在安装pytorch时极为恼火,明明电脑有GPU且pytorch已经装好,但是torch.cuda.is_available()一直是返回false。在网上搜集了一堆解决办法,最终摸索了两天后解决了这个问题。为避免下次遇到此问题,特此做个解决方法记录。
当出现torch.cuda.is_available()返回false的情况时解决办法

  1. 首先在自己创建的用于安装pytorch的虚拟环境中,输入conda list查看从官网下载的pytorch是CPU版本的还是GPU版本的。
  2. 若发现自己下载的pytorch为CPU版本的,则将此pytorch卸载,去官网使用PIP安装方式安装新的pytorch记住千万,千万,千万不要用conda的安装方式,因为它下载的一直都是CPU版本的
  3. 若自己下载的确实为GPU版本的,但是仍旧出现torch.cuda.is_available()返回false的情况,则可能是版本不匹配的问题。

安装pytorch时,torch.cuda.is_available()返回false的解决办法

原文:https://www.cnblogs.com/alex-jzw/p/15021956.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!