Pytorch 1.0 发布后还未提供conda或pip的安装方式,根据官网安装执行过程中发生了很多问题,特此记录。
官网地址:https://pytorch.org/get-started/locally/
官网上的命令行如下:
11 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,跑了很长时间还是没有成功,明天继续...
网友评论