ubuntu 安装cuda10.1

### 安装准备

为了确保顺利安装 CUDA 10.1，在开始之前需确认系统的准备工作已经完成。这包括更新系统软件包列表并升级已有的软件包到最新版本[^2]。

sudo apt update && sudo apt upgrade -y

### 添加 NVIDIA 软件源

对于稳定性和兼容性的考虑，建议先移除旧版本的NVIDIA驱动程序以及CUDA工具包。接着添加官方的NVIDIA PPA仓库来获取最新的驱动和支持：

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update

### 安装特定版本的 NVIDIA 驱动

选择合适的显卡驱动非常重要，因为不同的GPU可能需要不同版本的驱动才能正常工作。针对CUDA 10.1, 推荐使用与之匹配的418系列驱动:

sudo apt install nvidia-driver-418

重启计算机使新的图形驱动生效之后继续下面的操作。

### 下载 CUDA Toolkit 10.1

访问[NVIDIA官方网站](https://developer.nvidia.com/cuda-toolkit-archive)，找到对应于Ubuntu发行版的CUDA 10.1版本下载链接。考虑到网络速度因素，可以采用`axel`命令来进行加速下载[^4]:

sudo apt install axel
axel -n 10 https://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run

### 执行 CUDA Installer

赋予刚刚下载下来的`.run`文件执行权限，并运行它以启动安装向导。注意在安装过程中可以选择仅安装CUDA toolkit而不改变现有的驱动状态：

chmod +x cuda_*.run
sudo ./cuda_*.run --toolkit --silent --override

这里的`--silent`参数用于静默模式安装；而`--override`则允许覆盖任何潜在冲突。

### 设置环境变量

编辑用户的shell配置文件（如`.bashrc`），加入如下两行以便正确设置PATH和LD_LIBRARY_PATH环境变量指向新安装好的CUDA路径：

echo 'export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}' >> ~/.bashrc
source ~/.bashrc

此时可以通过输入`nvcc --version`验证CUDA是否被成功识别。

### cuDNN 库集成

前往[NVIDIA cuDNN页面](https://developer.nvidia.com/rdp/cudnn-archive)注册账号后下载适用于CUDA 10.1的cuDNN库压缩包。解压并将其中的内容复制至相应的CUDA目录下即可完成集成操作。

tar -xzvf cudnn-10.1-linux-x64-vX.Y.Z.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda-10.1/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64/
sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h /usr/local/cuda-10.1/lib64/libcudnn*

至此，整个CUDA 10.1及其配套组件的安装过程就全部结束了。