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不报错,则安装成功