栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

报错RuntimeError: torch.cuda.FloatTensor is not enabled.的可能原因及解决办法

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

报错RuntimeError: torch.cuda.FloatTensor is not enabled.的可能原因及解决办法

  使用pytorch时候报错RuntimeError: torch.cuda.FloatTensor is not enabled.,具体信息如下所示:

可能原因1

  出现这个错误,有可能是因为你的电脑没有英韦达显卡,不支持GPU加速,而你在代码中使用了gpu。解决办法就是把gpu版本的代码改成cpu版本的,也就是把代码里所有出现.cuda()的地方都修改为.cpu()即可。具体可以参考这里。

可能原因2

  还一个原因,刚好和上面的反过来了。当然,不是因为电脑没有cpu,而是你下载安装的pytorch是cpu版本的。而我出现这个错误的原因也正是如此。解决办法就是,将已经安装的pytorch卸载掉重装。由于我安装时候是自己下载的whl文件然后用pip安装的,所以卸载时候也比较简单,直接一行命令pip3 uninstall torch即可,运行后会出现确认信息,记得检查一下要卸载的版本是否正确(这里我一开始又弄错了,因为我使用conda管理了两个虚拟环境,刚开始没注意看确认信息,把我1.6.0版本的torch给卸载了,卸载完了一看好像不太对劲,然后才发现我没把终端的运行环境给切换过来,所以还是仔细一点)。
  卸载完成后,执行命令conda list,可以看到已经没有torch包了,这说明卸载成功了。然后就是重新下载安装torch,各版本torch的下载链接,里面有很多版本,注意仔细甄别,不要下cpu开头的,那个是cpu版本的torch。对于我要下载的torch0.4.0,它的名称为cu90/torch-0.4.0-cp36-cp36m-linux_x86_64.whl。
  下载好whl文件后,在该目录下打开终端,执行命令pip3 install torch-0.4.0-cp36-cp36m-linux_x86_64.whl进行安装即可。
  更详细的安装pytorch以及管理多虚拟环境可以参考这里。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/355434.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号