我是靠谱客的博主 不安酸奶,最近开发中收集的这篇文章主要介绍报错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以及管理多虚拟环境可以参考这里。

最后

以上就是不安酸奶为你收集整理的报错RuntimeError: torch.cuda.FloatTensor is not enabled.的可能原因及解决办法的全部内容,希望文章能够帮你解决报错RuntimeError: torch.cuda.FloatTensor is not enabled.的可能原因及解决办法所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(42)

评论列表共有 0 条评论

立即
投稿
返回
顶部