anaconda安装pytorch GPU版小白教程

step1:确认自己的cuda的版本号

nvcc --version

例如我的，显示结果如下：

由结果可以看出cuda版本号是9.0
step2:去官网查看命令
https://pytorch.org/

注意：如果cuda 不匹配，命令是无法使用的，比如我的cuda是9.0，则命令是：

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

step3：测试是否安装成功：

(pytorch-3) XXXX>python
Python 3.5.6 |Anaconda, Inc.| (default, Aug 26 2018, 16:05:27) [MSC v.1900 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> import torchvision

如import不报错，则安装成功