翻了一圈没有发现这个错误的优质解答,那就自己动手!在linux环境下,出现错误的原因是pytorch版本不匹配。笔者出错的原因是直接使用pip install torch造成的后果,会发现一些代码可以运行而另一些不行的问题(或者都不行)。
解决办法有两种,直接重装环境和卸载老版本的pytorch安装合适的版本。
重装环境的办法为,在命令行键入
conda create -n ___ python=3.6
其中__中填新环境的名称,可以任意填写,后面的python=3.6表示安装的版本是python3.6。
进入环境
source activate ___
其中__中填新环境的名称。
接着进入Pytorch官网找到合适的pytorch版本复制“Run this Command”的代码到命令行中,就可以下载啦。
光下载了pytorch还不行,一些安装包也是必备的,但是相对pytorch就容易一些,基本上可以直接用pip install 加需要下载的安装包名称就行。