Ubuntu用Anaconda 安装PyTorch、torchvision

适用于服务器上多用户安装于自己目录的情况

Anaconda

安装Anaconda见此篇

CUDA

安装CUDA见此篇

查看CUDA版本

nvcc -V

Pytorch & torchvision

修改软件源为清华的源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ 
conda config --set show_channel_urls yes 

去Pytorch官网根据需要的版本下载安装，需要关注的有：

• OS
• Package Manager
• Python
• CUDA

本次选择如图

1

按照command提示下载

conda install pytorch torchvision cuda90 -c pytorch

完成之后可在python中查看

import torch
import torchvision
print(torch.__version__)#输出PyTorch版本