Install CUDA & cuDNN on WSL2

[sko00o]

My Environment

• Win11 + WSL2 (Ubuntu 22.04 )

CUDA

Follow this official tutorial:

https://docs.nvidia.com/cuda/cuda-installation-guide-linux/#wsl

wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb

sudo apt-get update
sudo apt-get install cuda-toolkit

Environment Setup:

sudo cat <<\EOF >  /etc/profile.d/099_cuda.sh
export PATH="$PATH:/usr/local/cuda/bin"
export LD_LIBRARY_PATH=/usr/local/cuda/lib64\
                         ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
EOF

cuDNN

Go to https://developer.nvidia.com/cudnn-downloads .

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb

sudo apt-get install libcudnn9-cuda-12
sudo apt-get install libcudnn9-dev-cuda-12

Verify Installation

sudo apt-get install libcudnn9-samples

sudo apt install libfreeimage3 libfreeimage-dev

cd /tmp
cp -r /usr/src/cudnn_samples_v9.
cd cudnn_samples_v9/mnistCUDNN
make
./mnistCUDNN