安装显示驱动

• https://zhuanlan.zhihu.com/p/122286055

lspci|grep NVIDIA
lspci|grep -i nvidia


#测试GRID驱动是否安装成功:
nvidia-smi

下载CUDA toolkit

• https://developer.nvidia.com/cuda-downloads

• https://developer.nvidia.com/cuda-toolkit-archive

• 安装说明 https://blog.csdn.net/wang_yi_wen/article/details/78947860

  sudo chmod +x cuda_11.0.3_450.51.06_linux.run
  sudo ./cuda_11.0.3_450.51.06_linux.run
  

tensorflow和cuda对应表

• https://blog.csdn.net/Zinnir/article/details/122766636

pytorch和cuda对应版本安装

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