1、安装anaconda3:
https://www.anaconda.com/download/#windows
选Python3.6的x64版本下载、安装。
安装之后,把“C:\ProgramData\Anaconda3;C:\ProgramData\Anaconda3\Library\bin”加到系统环境变量(可选,主要为了后面运行conda命令)。
2、安装 CUDA。注意:需要有Nv显卡,否则没必要装了。目前最新的9.1版本:
https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10
下载network或local包都可以安装 。
3、下载 安装anaconda下的Pytorch包:
参考 蒲嘉宸 大神的文章:https://zhuanlan.zhihu.com/p/26871672
下载地址:https://pan.baidu.com/s/1dF6ayLr#list/path=%2F
根据CUDA是7、8、9选择 带有cu70、cu80、cu90的最新安装包下载。我选的是pytorch-0.3.1-py36_cuda90_cudnn7he774522_2.tar.bz2
下载完之后,运行:
cd "下载包的路径" conda install numpy mkl cffi conda install --offline pytorch????.tar.bz2
完成。可以运行代码进行测试。