Linux下CUDA环境的简单配置

这里主要是为了快速记录在Linux下关于Nvidia套件的配置工作

添加软件源

Nvidia已经为主流的操作系统准备了软件源，采用软件包管理器的方式安装会更加简单快捷

在这里进入CUDA下载页，选择对应的架构和操作系统

这里要注意的一点是，Installer Type要选择deb(network)便于采用在线安装，不然软件包可能会非常大

在线安装是可行的，因为如果用国内IP的话会自动将软件源域名定向到developer.download.nvidia.cn的国内节点，基本上能满速下载

选择好这几个选项后，底下就会出现安装流程，如果不想指定版本的话可以直接安装cuda-toolkit

Docker支持

在容器中使用CUDA支持是由Nvidia Container提供的，在这里需要额外安装，配置好上述软件源后，安装以下软件包

nvidia-container-toolkit

配置环境变量

安装完CUDA套件之后，诸如nvcc这类工具其实就已经安装好了，但是直接运行却提示找不到，这是因为CUDA套件的目录默认没有放在系统PATH底下，需要手动添加

不要去遵照当尝试执行nvcc时，软件包管理器给出的安装nvidia-cuda-toolkit建议，那是原来的公共镜像源里面的CUDA源，这样会导致环境混乱

编辑/etc/environment，将/usr/local/cuda/bin添加至PATH中，注意使用冒号分隔，一个典型的例子是这样

PATH="/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/games:/usr/local/games:/snap/bin:/usr/local/cuda/bin"

然后就能执行nvcc -V来查看配置情况