linux配置pytorch环境（conda）

遇到的问题

在import torchvision.io时候报错，表示没有io这个属性。检查了一下当时的torchvision版本是0.4.2，初步判断是torchvision的版本太低了。

之前安装torch和torchvision都是用的conda install torch类似的指令安装的，感觉安装的不太对，所以还是去官网找比较靠谱。

一、检查服务器环境

//1. 查看系统版本：
uname -a
//2. 查看cuda版本：
cat /usr/local/cuda/version.txt
//3. 查看conda版本：
conda -V
//4. python
python -V

我的环境：
OS：Ubuntu 16.04
CUDA：10.2.89
conda：4.5.11
python：3.7.7

二、conda常用指令

//1. 查看所有虚拟环境列表：
conda env list
//2. 激活某个虚拟环境：
source activate env_test1(环境名)
//3. 查看当前虚拟环境下的库：
conda list
//4. 退出当前虚拟环境
conda deactivate

三、安装

在pytorch官网找匹配的指令，主要是要和cuda匹配

用pip或者conda应该都是可以的，我用的pip。

pip3 install torch torchvision torchaudio

而且只要在激活的虚拟环境下（前面的括号里是激活的环境名），那么就会安装在这个虚拟环境下的：

四、验证

python
import torch
x = torch.rand(5, 3)
print(x)
torch.cuda.is_available()