ubuntu20.04 安装pytorch

在Ubuntu20.04系统中安装Pytorch，主要通过官方网址：​

https://pytorch.org/get-started/locally/​

通过选择不同的选项，然后复制最底下的命令，即可实现安装。​

如果是具有硬件显卡，例如Nvidia显卡，需要提前安装相关的驱动，CUDA等，之前曾参考相关的网络文章进行手动安装，例如文章 https://blog.csdn.net/ashome123/article/details/105822040/​

后来通过命令行 ubuntu-drivers autoinstall 直接安装即可。如果此前曾经安装过其他版本的，则可以提前用apt remove \*nvidia\* 来卸载相关插件然后重新安装。​

查询cuda版本的方法：nvcc -V 或者是nvidia-smi查看。​

安装完成后，可以测试是否安装成功：​

import torch
torch.__version__ 
'1.10.2+cu113' 
torch.cuda.is_available() 
True

​​