Skip to content

Instantly share code, notes, and snippets.

@shkarupa-alex
Last active March 22, 2024 06:33
Show Gist options
  • Save shkarupa-alex/4829bc79a1bef8740ddaedd809f68c40 to your computer and use it in GitHub Desktop.
Save shkarupa-alex/4829bc79a1bef8740ddaedd809f68c40 to your computer and use it in GitHub Desktop.
Ubuntu 22.04 + Cuda
# Prioritize NVIDIA packages
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
# Fetch NVIDIA keys
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/3bf863cc.pub
# Add NVIDIA repos
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/ /"
sudo apt-get update && sudo apt-get upgrade
# Install NVIDIA driver & CUDA
# + stable for RTX 4090 + TF 2.13
sudo apt-get install cuda-drivers cuda-11-8 libcudnn8=8.9.2.26-1+cuda11.8 libcudnn8-dev=8.9.2.26-1+cuda11.8 libnccl2 libnccl-dev
# - latest for TF 2.16.1
# sudo apt-get install cuda-drivers cuda-12-2 libcudnn8 libcudnn8-dev libnccl2 libnccl-dev
# Reboot. Check that GPUs are visible
nvidia-smi
@natalisuh
Copy link

thank you!
in a 3rd line typo 2004-2204

@shkarupa-alex
Copy link
Author

thank you! in a 3rd line typo 2004-2204

fixed

@mintisan
Copy link

It will encounter nvidia-dkms-535 : Depends: nvidia-kernel-common-535 (= 535.129.03-0ubuntu1) but 535.129.03-0ubuntu0.22.04.1 is to be installed problems.

And we can solve with https://askubuntu.com/questions/1491916/problem-with-nvidia-package-dependencies .

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment