部署CUDA

使用PuTTY工具，以root用户登录服务器。
查看nouveau驱动是否已禁用。
```
lsmod |grep nouveau
```
回显不为空即未禁用，继续3的操作。
禁用nouveau驱动。
1. 新建文件“/etc/modprobe.d/disable-nouveau.conf”。
```
vim /etc/modprobe.d/disable-nouveau.conf
```
2. 添加如下两行。
```
blacklist nouveau
options nouveau modeset=0
```

备份并生成新的initramfs文件。

cp /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r)-nouveau.img
dracut -f /boot/initramfs-$(uname -r).img $(uname -r)

重启机器。
```
reboot
```
确认nouveau驱动已禁用。
```
lsmod |grep nouveau
```
回显为空即已禁用。

安装NVIDIA-CUDA组件。

wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux_sbsa.run
sudo sh cuda_11.4.0_470.42.01_linux_sbsa.run

配置环境变量。

export CUDA_HOME=/usr/local/cuda
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH

验证安装是否成功。
```
nvcc -v
```
回显显示版本信息即为安装成功。

父主题： 软件栈部署