Ubuntu16.04+CUDA9.0安装pytorch0.4.1填坑记录

直接说解决方法吧。

你可能试过直接用conda install 但是无法使用清华源

你也可能试过下载清华源的安装包，然后安装，但是在安装其他依赖的时候却提示你升级（升级包是pytorch的）

正确安装的姿势是：

1.首先到pytorch官网下载对应版本的whl文件

下载完成后，用pip install 文件名 来安装pytorch

2.使用conda install torchvision安装torchvison。此时anaconda会提示你下载其他依赖包，但是你会发现：

下载的是清华源的，下载完成后，就OK了。

3.接下来就是测试啦

你只要输入

import torch

import torchvision

torch.cuda.is_available()

如果输出的是True

就OK了！