1.Anaconda很好装，直接过

2.Nvidia驱动安装，可以到官网找到驱动版本与CUDA版本的对应号，尽量谨慎选择驱动，因为在Ubuntu下Nvidia驱动的坑非常多

3.根据驱动选择对应的CUDA版本，下载后只安装Toolkit，其它的不是必须的，并将对应的CUDNN文件放入CUDA中

4.安装根据toolkit版本安装PYTORCH

ubuntu20.04中Pytorch配置踩坑