1.Anaconda很好装，直接过
2.Nvidia驱动安装，可以到官网找到驱动版本与CUDA版本的对应号，尽量谨慎选择驱动，因为在Ubuntu下Nvidia驱动的坑非常多
3.根据驱动选择对应的CUDA版本，下载后只安装Toolkit，其它的不是必须的，并将对应的CUDNN文件放入CUDA中
4.安装根据toolkit版本安装PYTORCH
ubuntu20.04中Pytorch配置踩坑
原文：https://www.cnblogs.com/lanqinglu/p/14627049.html
