linux安装pytorch-GPU版本

1. 查看cuda版本：

ls -l /usr/local | grep cuda

得到如下结果：

lrwxrwxrwx  1 root root   21 12月 20  2019 cuda -> /usr/local/cuda-10.2/
drwxr-xr-x 18 root root 4096 12月 20  2019 cuda-10.2

说明cuda是10.2版本

2. conda安装pytorch-GPU

首先参考官网的命令：https://pytorch.org/get-started/locally/

给的命令是：conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

但是实际由于魔法的存在，不能直接安装，所以需要先更换pytorch源，请参考：https://blog.csdn.net/weixin_35757704/article/details/108910070（Mac、linux相同）

安装命令是：conda install pytorch torchvision torchaudio cudatoolkit=10.2，这里要把-c pytorch删掉！

然后安装即可

3. 测试

打开代码，运行：

import torch

print(torch.cuda.is_available())