Linux pytorch安装 cuda，torch，torchvision版本选择

安装好anaconda后，安装pytorch：

（1）使用pip install** ，代替conda install***更方便

（2）需要用到GPU资源的话，要加cuda的内容。linux查看本机cudu版本的命令为：cat /usr/local/cuda/version.txt

（3）知道了cuda的版本，去查看对应torch和torchvision的版本

 

 我的cuda版本是10.0.130，所以我要安装的pytorch版本是1.1，1.2这些。我选择1.2版本，那torch对应也是1.2版本，torchvision就选择0.4.0版本。

（4）直接在shell里输入：pip install torch==1.2.0 torchvison==0.4.0，回车，等他运行结束

（5）shell键入“python”进入python环境，然后输入“import torch"，看看是否成功，”print(torch.cuda.is_available()) ”查看cuda版本是否对应。如果没有按照上面的版本对应的话，他会返回“False”，对应上了，就返回“True"。