AndyL
February 4, 2025, 6:05am
1
I’m running NX on Ubuntu 2204, but struggling to find the right CUDA/CUDNN drivers for Ubuntu 2404. (when compiling my elixir code I get c_src/exla/exla_cuda.cc:4:10: fatal error: cuda_runtime.h: No such file or directory
)
Can anyone post an CUDA/CUDNN install script that works on Ubuntu 2404?
AndyL
February 6, 2025, 1:41am
2
Click here to find a working install script.
I can help with that! Here’s a script to install CUDA and cuDNN on Ubuntu 24.04:
#!/bin/bash
# Step 1: Disable Wayland and Nouveau
sudo bash -c 'echo "WaylandEnable=false" >> /etc/gdm3/custom.conf'
sudo bash -c 'echo "blacklist nouveau" >> /etc/modprobe.d/blacklist.conf'
sudo bash -c 'echo "blacklist nvidia" >> /etc/modprobe.d/blacklist.conf'
sudo bash -c 'echo "blacklist nvidia-drm" >> /etc/modprobe.d/blacklist.conf'
sudo bash -c 'echo "blacklist nvidia-modeset" >> /etc/modprobe.d/blacklist.conf'
sudo update-initramfs -u
sudo reboot
# Step 2: Add NVIDIA PPA and Install NVIDIA Driver
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
sudo apt install nvidia-driver-555 -y
sudo reboot
# Step 3: Install CUDA Toolkit
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-repo-ubuntu2404_12.5.1-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2404_12.5.1-1_amd64.deb
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/7fa2af80.pub
sudo apt update
sudo apt install cuda -y
# Step 4: Install cuDNN
wget https://developer.download.nvidia.com/compute/cudnn/9.2.1/local_installers/cudnn-local-repo-ubuntu2404-9.2.1_1.0-1_amd64.deb
sudo dpkg -i cudnn-local-repo-ubuntu2404-9.2.1_1.0-1_amd64.deb
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cudnn/repos/ubuntu2404/x86_64/7fa2af80.pub
sudo apt update
sudo apt install libcudnn9 -y
# Step 5: Setup Environment Variables
echo 'export PATH=/usr/local/cuda-12.5/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-12.5/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
echo 'export CUDNN_INCLUDE_DIR=/usr/include' >> ~/.bashrc
echo 'export CUDNN_LIB_DIR=/usr/lib/x86_64-linux-gnu' >> ~/.bashrc
source ~/.bashrc
# Step 6: Verify Installations
nvidia-smi
nvcc --version
ls /usr/include/cudnn*.h
ls /usr/lib/x86_64-linux-gnu/libcudnn*
echo "Installation complete!"
This script should help you get CUDA and cuDNN up and running on Ubuntu 24.04. Let me know if you run into any issues or need further assistance!