Ubuntu22.04 : NVIDIA GeForce RTX 4090 v535.98, cuda 12.0, cuDNN v8.9.3

• cat /etc/issue
  Ubuntu 22.04.3 LTS \n \l

• lspci | grep -i nvidia
  01:00.0 VGA compatible controller: NVIDIA Corporation Device 2684 (rev a1)
  01:00.1 Audio device: NVIDIA Corporation Device 22ba (rev a1)
  41:00.0 VGA compatible controller: NVIDIA Corporation GK208B [GeForce GT 710] (rev a1)
  41:00.1 Audio device: NVIDIA Corporation GK208 HDMI/DP Audio Controller (rev a1)

• Install gcc
  sudo apt install gcc

• Install make
  sudo apt install make

• Forbiden nouveau before installing Nvidia Driver
  sudo vim /etc/modprobe.d/blacklist.conf

  blacklist nouveau
  blacklist lbm-nouveau
  options nouveau modeset=0
  alias nouveau off
  alias lbm-nouveau off
  

• Update initramfs
  sudo update-initramfs -u
  reboot

• Validate
  lsmod | grep nouveau

• Download Nvidia Driver
  
  

sudo chmod +x NVIDIA-Linux-x86_64-535.98.run
sudo bash ./NVIDIA-Linux-x86_64-535.98.run

• Validate Nvidia Driver

• CUDA Toolkit 12.0 Downloads

wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run
sudo sh cuda_12.2.0_535.54.03_linux.run

sudo vim ~/.bashrc , add the belows:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

• Validate CUDA
  nvcc -V
  

• cuDNN
  

• uncompress cudnn file

  tar -xf cudnn-linux-x86_64-8.9.3.28_cuda12-archive.tar.xz

• install cudnn to real path

sudo cp cudnn-linux-x86_64-8.9.3.28_cuda12-archive/include/cudnn.h /usr/local/cuda-12.2/include
sudo cp cudnn-linux-x86_64-8.9.3.28_cuda12-archive/lib/libcudnn* /usr/local/cuda-12.2/lib64
sudo chmod a+r /usr/local/cuda-12.2/include/cudnn.h /usr/local/cuda-12.2/lib64/libcudnn*

• install Anaconda

sudo bash Anaconda3-2022.10-Linux-x86_64.sh
source ~/.bashrc

Create virtual environment

conda create -n dsg(replace with your name) python=3.8

Activate virtual environment

conda activate dsg (replace with your name)

• install pytorch

  pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121

• test

python  # Enter python Environment
import torch  # import pytorch package
torch.cuda.is_available()  # check if  cuda is available,True means available,False means unavailable.