在anaconda上安装指定版本的pytorch

2020-07-14  本文已影响0人  全都给我Pass

在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

窗口截图如下:

上一篇下一篇

猜你喜欢

热点阅读