在 Ubuntu 20.04 安装感兴趣的 Pytorch 版本

先到官网首页查看感兴趣版本的pytorch对应的cuda版本，

1. 安装对应版本的cuda 和 cudnn

2. 根据 pytorch 官网首页的命令安装pytorch

3.验证：

$ ipytorch3
In [2]: import torch

In [3]: torch.cuda.is_available
Out[3]: <function torch.cuda.is_available() -> bool>

In [4]: torch.cuda.is_available()
Out[4]: True

In [5]: a=torch.tensor()
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-5-ad45ed793f89> in <module>
----> 1 a=torch.tensor()

TypeError: tensor() missing 1 required positional arguments: "data"

In [6]: a=torch.tensor(3)

In [7]: a
Out[7]: tensor(3)

In [8]: a.device
Out[8]: device(type='cpu')

In [9]: a=a.cuda()

In [10]: a.device
Out[10]: device(type='cuda', index=0)

In [11]: exit
opencl@opencl-PC:~$

pytorch 和cuda 皆可用