美文网首页
Pytorch 1.0 在Windows系统的安装

Pytorch 1.0 在Windows系统的安装

作者: 苏见 | 来源:发表于2018-12-05 18:32 被阅读201次

    Pytorch 1.0 发布后还未提供conda或pip的安装方式,根据官网安装执行过程中发生了很多问题,特此记录。

    官网地址:https://pytorch.org/get-started/locally/

    官网上的命令行如下:

      1

    1 git clone pytorch

    clone该项目本身没问题,但是clone一些第三方项目的时候可能会出错,clone完发现项目内third_party目录下很多文件夹里是空的,只有一个.git文件夹,cmd进入这些目录执行git status发现文件都被删除了,执行git checkout --后文件回来了。

    2 第二行

    cd pytorchset "VS150COMNTOOLS=C:\Program Files (x86)\Microsoft Visual Studio\2017\Enterprise\VC\Auxiliary\Build"

    这其实是两步,合并到一行了,应该分步骤执行

    cd pytorch

    set "VS150COMNTOOLS=C:\Program Files (x86)\Microsoft Visual Studio\2017\Enterprise\VC\Auxiliary\Build"

    另外,我本地安装的visual studio 2017不是Enterprise版本,而是Community版本,所以目录改为:

    set "VS150COMNTOOLS=C:\Program Files (x86)\Microsoft Visual Studio\2017\Community\VC\Auxiliary\Build"

    如果是Visual Studio 2015,则改为:

    set "VS140COMNTOOLS=C:\Program Files (x86)\Microsoft Visual Studio 14.0\VC"

    3 第三、四行

    在cmd下直接执行

    4 第六、七行

    只有python2.7版本需要执行

    5 第九、十行

    针对CUDA 8, VS2015 Update 3需要执行

    6 第十二行

    call "%VS150COMNTOOLS%\vcvarsall.bat" x64 -vcvars_ver=14.11

    VS150COMNTOOLS就是第二步设置的环境变量,也可以用everything 找出vcvarsall.bat文件的目录,替换该变量。

    vcvars_ver查看,打开Visual Studio,点击菜单栏“工具-获取工具和功能”,将会打开组件安装窗口,点击"windows-使用c++的桌面开发",可看到右侧的安装详细信息,可选项中有VC++ 2017版本15.8 v14.15最新v141工具,如下图,vcvars_ver就是14.15。

    所以这条命令我改成了:

    call "C:\Program Files (x86)\Microsoft Visual Studio\2017\Community\VC\Auxiliary\Build\vcvarsall.bat" x64 -vcvars_ver=14.15

    7 最后一行

    直接执行python setup.py install,跑了很长时间还是没有成功,明天继续...

    相关文章

      网友评论

          本文标题:Pytorch 1.0 在Windows系统的安装

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