1、安装 anaconda
Python 3.6.2 |Anaconda 4.3.0 (64-bit)| (default, Jul 20 2017, 12:30:02) [MSC v.1900 64 bit (AMD64)]
https://www.anaconda.com/download/官网下载anaconda安装包Anaconda3-5.0.1-Windows-x86_64.exe
2、安装cuda8.0+cudnn8.0
NVIDIA官网下载cuda_8.0.61_win10.exe
和cudnn-8.0-windows10-x64-v6.0.zip
3、安装numpy等包
打开anaconda navigator->environment->my_root_>open terminal
输入
conda install numpy mkl cffi
等待安装完成。
4、安装pytorch
conda install --offline pytorch-0.2.1-py36he6bf560_0.2.1cu80.tar.bz2
等待安装完成。
测试:打开anaconda navigator->environment->my_root_>open python
输入
import torch
没有报错,表示安装成功。