在pycharm的终端terminal上不好安装pytorch
还是用回像之前安装tensorflow一样的老方法去安装pytorch比较高成功率,
由于tf还是pytorch的版本和cuda python版本都有一一对应的关系,
因此最好的方法还是找到对应的所有包版本捆绑好的.whl文件安装。
步骤:
到网址:https://download.pytorch.org/whl/torch_stable.html下载对应我电脑cudn python环境的whl文件,
由于我的python 是3.6版本,cuda 是10.0版本,windows10。需要下载的pytorch为1.3.1,因此对应的whl文件是:
torch-1.3.1+cpu-cp36-cp36m-win_amd64.whl,
把这个文件下载到本地目录:C:\install\torch-1.3.1+cpu-cp36-cp36m-win_amd64.whl。
然后打开anaconda prompt,进入(激活)你项目指定运行的虚拟环境里,我这叫tensorflow(名字关系不大,由于这个虚拟环境下安装的库比较少,少会出现冲突,由不想重新建一个,因此沿用它了)。
常用conda命令:
显示现有的所有虚拟环境:conda env list
Windows激活虚拟环境:[conda] activate env_name
进入到whl文件所在目录去安装指定版本1.3.1的pytorch:pip install C:\install\torch-1.3.1+cpu-cp36-cp36m-win_amd64.whl -ihttps://pypi.tuna.tsinghua.edu.cn/simple
窗口截图如下: