安装CUDA

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

使用deb安装的话，有时会报错：

dpkg: 处理软件包 nvidia-driver-450 (--configure)时出错：
 依赖关系问题 - 仍未被配置
dpkg: 依赖关系问题使得 cuda-drivers-450 的配置工作不能继续：
 cuda-drivers-450 依赖于 nvidia-dkms-450 (>= 450.51.05)；然而：
  软件包 nvidia-dkms-450 尚未配置。
 cuda-drivers-450 依赖于 nvidia-driver-450 (>= 450.51.05)；然而：
  软件包 nvidia-driver-450 尚未配置。

解决办法：

sudo apt clean
sudo apt-get update
sudo apt purge nvidia-*
sudo apt autoremove

最后下载run文件进行安装，把第一项取消，然后install

配置环境

vim ~/.bashrc
export PATH=/usr/local/cuda-11.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH
source ~/.bashrc

确认

$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Oct_12_20:09:46_PDT_2020
Cuda compilation tools, release 11.1, V11.1.105
Build cuda_11.1.TC455_06.29190527_0