Win10配置pytorch0.4.0踩坑,CUDA9.1+CUDNN7.1+python3.6
在Win10上配置Pytorch0.4.0
1.准备工作
1.1安装Visual Studio 2017
安装Visual Studio 2017的免费的Community版即可
安装过程中需要选中工作负载中的[使用C++桌面开发]和单个组件中的VC++ 2017版本 v14.11工具箱
1.2 安装Anaconda3
下载Anaconda3,下载安装python3.6对应的64位版本即可
Python3.5版本编译pytorch未通过
1.3 安装CUDA9.1和CUDNN7.1
将CUDNN中bin、include、lib文件夹中文件直接复制到“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1”路径下的bin、include、lib文件夹中
2 安装Pytorch
2.1 下载源码
git clone --recursive https://github.com/pytorch/pytorch
若直接Download源码,第三方库无法下载到,需要手动逐一下载并放放置到相应位置
2.2 安装依赖库
打开Anaconda Prompt
conda install numpy pyyaml mkl mkl-include setuptools cmake cffi typing
2.3 编译
#cd到pytorch源码文件夹下
cd pytorch
xcopy /Y aten\src\ATen\common_with_cwrap.py tools\shared\cwrap_common.py
#安装vs2017免费Community版,则需要把Enterprise修改为Community
set "VS150COMNTOOLS=C:\Program Files (x86)\Microsoft Visual Studio\2017\Enterprise\VC\Auxiliary\Build"
set CMAKE_GENERATOR=Visual Studio 15 2017 Win64
set DISTUTILS_USE_SDK=1
call "%VS150COMNTOOLS%\vcvarsall.bat" x64 -vcvars_ver=14.11
python setup.py install
至此安装完毕,关闭Anaconda Prompt,重新打开新的cmd窗口
python
import torch
成功