Install CUDA 11.2, cuDNN 8.1.0, PyTorch v1.8.0 (or v1.9.0), and python 3.9 on RTX3090 for deep learning

This tutorial is tested with RTX3090 on Ubuntu 20.04 LTS. The refereed installation guide is from: https://medium.com/@dun.chwong/the-ultimate-guide-ubuntu-18-04-37bae511efb0

1. Configuration

Ubuntu setup by running the following commands.

sudo apt-get update
sudo apt-get upgrade -y
sudo apt-get install -y build-essential cmake unzip pkg-config
sudo apt-get install -y libxmu-dev libxi-dev libglu1-mesa libglu1-mesa-dev
sudo apt-get install -y libjpeg-dev libpng-dev libtiff-dev
sudo apt-get install -y libavcodec-dev libavformat-dev libswscale-dev libv4l-dev
sudo apt-get install -y libxvidcore-dev libx264-dev
sudo apt-get install -y libgtk-3-dev
sudo apt-get install -y libopenblas-dev libatlas-base-dev liblapack-dev gfortran
sudo apt-get install -y libhdf5-serial-dev graphviz
sudo apt-get install -y python3-dev python3-tk python-imaging-tk
sudo apt-get install -y linux-image-generic linux-image-extra-virtual
sudo apt-get install -y linux-source linux-headers-generic

2. Nvidia (Driver, CUDA, cuDNN)

2.1 Install Nvidia Driver

Step 1: Remove existing Nvidia drivers if any

sudo apt-get purge nvidia*

Step 2: Add Graphic Drivers PPA

sudo add-apt-repository…