在Windows安装源码安装PyTorch 0.4。

Windows 10 Home Visual Studio 2017 Community CUDA 9.0 + cuDNN 7.0 Anaconda 5,Python 3.6.4

1、安装软件

1.1、安装Visual Studio

需要安装Visual Studio 2017，免费的Community版是可以的，不要预览版，正式版即可。打开Visual Studio Installer，需要安装： - 使用C++桌面开发 - VC++ 2017版本 v14.11工具箱 - Anaconda也可以勾选，这样你就可以跳过安装Anaconda这一步

安装即可。

1.2、安装Anaconda

下载Anaconda,Python3版本。 安装，勾选添加到PATH，没有勾选的话手动添加吧！ 

1.3、安装CUDA和cuDNN

没有英伟达显卡或者不需要英伟达显卡支持跳过，安装时需要设置环境变量NO_CUDA=1。 我安装的是CUDA 9.0、cuDNN 7.0 需要勾选Nsight Visual Studio Edition。 

2、安装PyTorch

2.1、前期准备

2.1.1 修改CUDA头文件

最新版本的Visual Studio 2017,安装时会报错,修改CUDA头文件即可: unsupported Microsoft Visual Studio version! Only the versions 2012, 2013, 2015 and 2017 are supported! 不需要CUDA支持跳过

打开 - %CUDA_PATH%\include\crt\host_config.h -

#if _MSC_VER < 1600 || _MSC_VER > 1911//把1911 改为1920即可，保存

2.1.2 安装依赖

• conda设置清华大学镜像，比较快。安装pytorch 0.3也可以通过清华大学镜像

conda create -n pytorch python=3.6 anaconda#这步不是必须的，但推荐新建一个，这样子安装失败也不会影响自己已经配置好的环境activate pytorch 
#这个命令要在CMD下才有效，不知道为什么PowerShell无效#建议此后的命令都通过CMDconda install numpy pyyaml mkl mkl-include setuptools cmake cffi typing

2.1.3、下载PyTorch源代码

#保持网络比较好即可，学校网络差的要死，clone几次还不成功，最后只能用流量开个热点git clone --recursive https://github.com/pytorch/pytorch
cd pytorch

2.2、安装

xcopy /Y aten\src\ATen\common_with_cwrap.py tools\shared\cwrap_common.pyset "VS150COMNTOOLS=D:\ProgramData\MicroSoft\VisualStudio\Community\VC\Auxiliary\Build" #上面目录需要替换为你自己Visual Studio安装目录set CMAKE_GENERATOR=Visual Studio 15 2017 Win64set DISTUTILS_USE_SDK=1

call "%VS150COMNTOOLS%\vcvarsall.bat" x64 -vcvars_ver=14.11#14.13等其他版本成不成功没试过python setup.py install

等一段时间，大功告成。我安装过程中出错就第一次忘记改CUDA的头文件引起的。

2.3、测试

先退出安装的那个CMD，重新打开一个CMD或者PowerShell。 

参考：https://github.com/pytorch/pytorch

转载声明：本文转载自异步社区

原文链接：https://www.epubit.com/articleDetails?id=N4e159729-b5fb-4867-9f17-d2289deab318