概述
使用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.的可能原因及解决办法所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
发表评论 取消回复