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了!