美文网首页
在anaconda上安装指定版本的pytorch

在anaconda上安装指定版本的pytorch

作者: 全都给我Pass | 来源:发表于2020-07-14 15:53 被阅读0次

    在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

    窗口截图如下:

    相关文章

      网友评论

          本文标题:在anaconda上安装指定版本的pytorch

          本文链接:https://www.haomeiwen.com/subject/ybmuwxtx.html