- 参考官方文档安装cuda:
# wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-ubuntu2404.pin
# sudo mv cuda-ubuntu2404.pin /etc/apt/preferences.d/cuda-repository-pin-600
# sudo dpkg -i cuda-repo-ubuntu2404-12-6-local_12.6.3-560.35.05-1_amd64.deb
# sudo cp /var/cuda-repo-ubuntu2404-12-6-local/cuda-*-keyring.gpg /usr/share/keyrings/
# sudo apt-get update
# sudo apt-get -y install cuda-toolkit-12-6
- 配置cuda到环境变量中,似的命令行对nvcc等命令生效
# vi /etc/profile.d/apps-bin-path.sh
cuda_path="/usr/local/cuda-12.6/bin" if [ -n "${PATH##*${cuda_path}}" ] && [ -n "${PATH##*${cuda_path}:*}" ]; then export PATH="$PATH:${cuda_path}" fi |
# source /etc/profile
- 检查生效与否:
# env |grep PATH

# nvcc --version

- 安装cudnn

sudo dpkg -i cudnn-local-repo-ubuntu2404-9.6.0_1.0-1_amd64.deb
sudo cp /var/cudnn-local-repo-ubuntu2404-9.6.0/cudnn-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cudnn
---END