安装CUDA
操作步骤
- 使用PuTTY工具,以root用户登录服务器。
- 执行以下命令查看Nouveau驱动是否已禁用。
lsmod |grep nouveau
- 若有回显信息,表示Nouveau驱动未禁用,则执行3。
- 若无回显信息,表示Nouveau驱动已禁用,则结束操作。
- 执行以下命令禁用Nouveau驱动。
- 新建文件“disable-nouveau.conf”。
vi /etc/modprobe.d/disable-nouveau.conf
- 按“i”进入编辑模式,添加如下内容。
blacklist nouveau options nouveau modeset=0
- 按“Esc”键,输入wq!,按“Enter”保存并退出编辑。
- 新建文件“disable-nouveau.conf”。
- 执行以下命令备份并生成新的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
回显显示如下信息,表示安装成功。
父主题: 配置编译环境