Pytorch 1.0 在Windows系统的安装

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系统的安装)