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

????最近在安装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的解决办法

上一篇:单元测试 节选自《闻缺陷则喜》(此书可免费下载)


下一篇:Git