一、创建conda环境
conda create -n pytorch python=3.6 anaconda
二、确定cuda版本
nvcc -V
我使用的版本是9.0
三、安装PyTorch
用的是官网的安装命令
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
成功安装
四、验证安装
命令行输入python
>>import torch
>>improt torchvision
>>torch.cuda.is_available()
不报错，返回True，配置成功
五、可能出现的问题
如果第四步返回的是False，则需要看下是不是驱动和cuda的版本对应不上，在命令行输入nvidia-smi
384对应的就是9.0，387对应的是9.1，其他的对应关系自己可以查询下，找到正确对应版本后，很大几率可以配置成功。
