Ubuntu 22.04 LTS AIGC GPU环境搭建(NVIDIA 550.67+CUDA 12.4+cuDNN)
以下是在Ubuntu 22.04 LTS上安装NVIDIA驱动、CUDA Toolkit 12.4和cuDNN的步骤:
- 更新系统包索引并升级现有的包:
sudo apt update
sudo apt upgrade
- 安装NVIDIA驱动:
sudo apt install nvidia-driver-550.67
- 重启电脑以确保NVIDIA驱动正确安装:
sudo reboot
- 验证NVIDIA驱动安装情况:
nvidia-smi
下载CUDA Toolkit 12.4:
前往NVIDIA官网下载CUDA Toolkit 12.4(https://developer.nvidia.com/cuda-downloads),选择对应的安装包。
- 安装CUDA Toolkit 12.4:
sudo sh cuda_12.4.0_521.15_linux.run
配置环境变量:
编辑
~/.bashrc
或~/.profile
文件,添加以下内容:
export PATH=/usr/local/cuda-12.4/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.4/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
- 重新加载环境变量:
source ~/.bashrc
- 验证CUDA安装:
nvcc --version
下载cuDNN库:
前往NVIDIA官网下载与CUDA 12.4对应的cuDNN库(https://developer.nvidia.com/cudnn)。
- 安装cuDNN库:
tar -xzvf cudnn-linux-x86_64-8.4.0.19_cuda12-archive.tar.gz
sudo cp cuda/include/cudnn*.h /usr/local/cuda-12.4/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-12.4/lib64
sudo chmod a+r /usr/local/cuda-12.4/include/cudnn*.h /usr/local/cuda-12.4/lib64/libcudnn*
以上步骤安装了NVIDIA驱动、CUDA Toolkit 12.4和cuDNN,为AIGC等GPU加速应用提供了必要环境。
评论已关闭