Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save Clivern/fea4cb1aaf6a1e9897485ad54a73ad14 to your computer and use it in GitHub Desktop.
Save Clivern/fea4cb1aaf6a1e9897485ad54a73ad14 to your computer and use it in GitHub Desktop.

Ubuntu 22.04 for Deep Learning

In the name of God

This gist contains steps to setup Ubuntu 22.04 for deep learning.


Install Ubuntu 22.04

  • Computer name: Name-PC
  • Name: Name
  • User name: name
  • Password: ********

Update Ubuntu

$ sudo apt update
$ sudo apt full-upgrade --yes
$ sudo apt autoremove --yes
$ sudo apt autoclean --yes
$ reboot

Create Update Script

  • Create a file (~/full-update.sh) with the following lines:
#!/bin/bash

if [ "$EUID" -ne 0 ]
  then echo "Error: Please run as root."
  exit
fi

clear

echo "################################################################################"
echo "Updating list of available packages..."
echo "--------------------------------------------------------------------------------"
apt update
echo "################################################################################"
echo

echo "################################################################################"
echo "Upgrading the system by removing/installing/upgrading packages..."
echo "--------------------------------------------------------------------------------"
apt full-upgrade --yes
echo "################################################################################"
echo

echo "################################################################################"
echo "Removing automatically all unused packages..."
echo "--------------------------------------------------------------------------------"
apt autoremove --yes
echo "################################################################################"
echo

echo "################################################################################"
echo "Clearing out the local repository of retrieved package files..."
echo "--------------------------------------------------------------------------------"
apt autoclean --yes
echo "################################################################################"
echo

Change Settings

  • Review Ubuntu Settings

Change Software & Updates

  • Review Software & Updates

Update Ubuntu

  • $ sudo ~/full-update.sh

  • $ sudo dpkg -i google-chrome-stable_current_amd64.deb

Install Development Tools

  • $ sudo apt install build-essential pkg-config cmake cmake-qt-gui ninja-build valgrind

Install Python 3

  • $ sudo apt install python3 python3-wheel python3-pip python3-venv python3-dev python3-setuptools

Install Git

$ sudo apt install git
$ git config --global user.name "Name"
$ git config --global user.email "name@domain.com"
$ git config --global core.editor "gedit -s"
  • Copy your own SSH keys to ~/.ssh/

Install NVIDIA Drivers for Deep Learning

Check Display Hardware:

  • $ sudo lshw -C display

Install NVIDIA GPU Driver:

  • Install from GUI: Software & Updates > Additional Drivers > NVIDIA

Try $ sudo ubuntu-drivers autoinstall if NVIDIA drivers are disabled.

You can also install it from the terminal: $ sudo apt install nvidia-driver-535.

Check TensorFlow and CUDA Compatibilities:

Install CUDA Toolkit (CUDA 11.8):

  1. Install prerequisites:
    • $ sudo apt install linux-headers-$(uname -r)
  2. Download CUDA 11.8 (https://developer.nvidia.com/cuda-toolkit-archive)
    • $ wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
  3. Install CUDA 11.8: $ sudo ./cuda_11.8.0_520.61.05_linux.run --override (without Driver)
  4. Set up the development environment by modifying the PATH and LD_LIBRARY_PATH variables (Add following lines to ~/.bashrc):
    • export PATH=$PATH:/usr/local/cuda-11.8/bin
    • export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64:/usr/local/cuda-11.8/extras/CUPTI/lib64
  5. Check that GPUs are visible using the following command (A reboot may be required):
    • $ nvidia-smi

Install cuDNN v8.6, for CUDA 11.8:

Reboot:

  • $ reboot

Machine Learning Environment

$ python3 -m venv ~/venvs/ml
$ source ~/venvs/ml/bin/activate
(ml) $ pip install --upgrade pip setuptools wheel
(ml) $ pip install --upgrade numpy scipy matplotlib ipython jupyter pandas sympy nose
(ml) $ pip install --upgrade scikit-learn scikit-image
(ml) $ deactivate

Deep Learning Environment (TensorFlow-CPU)

$ python3 -m venv ~/venvs/tfcpu
$ source ~/venvs/tfcpu/bin/activate
(tfcpu) $ pip install --upgrade pip setuptools wheel
(tfcpu) $ pip install --upgrade opencv-python opencv-contrib-python
(tfcpu) $ pip install --upgrade tensorflow-cpu tensorboard keras
(tfcpu) $ deactivate

Deep Learning Environment (TensorFlow-GPU)

$ python3 -m venv ~/venvs/tfgpu
$ source ~/venvs/tfgpu/bin/activate
(tfgpu) $ pip install --upgrade pip setuptools wheel
(tfgpu) $ pip install --upgrade opencv-python opencv-contrib-python
(tfgpu) $ pip install --upgrade tensorflow tensorboard keras
(tfgpu) $ deactivate

Verification:

$ source ~/venvs/tfgpu/bin/activate
(tfgpu) $ python
>>> from tensorflow.python.client import device_lib
>>> device_lib.list_local_devices()
>>> exit()
(tfgpu) $ deactivate

Deep Learning Environment (PyTorch-CPU)

$ python3 -m venv ~/venvs/torchcpu
$ source ~/venvs/torchcpu/bin/activate
(torchcpu) $ pip install --upgrade pip setuptools wheel
(torchcpu) $ pip install --upgrade opencv-python opencv-contrib-python
(torchcpu) $ pip install --upgrade torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu
(torchcpu) $ deactivate

Deep Learning Environment (PyTorch-GPU)

$ python3 -m venv ~/venvs/torchgpu
$ source ~/venvs/torchgpu/bin/activate
(torchgpu) $ pip install --upgrade pip setuptools wheel
(torchgpu) $ pip install --upgrade opencv-python opencv-contrib-python
(torchgpu) $ pip install --upgrade torch torchvision torchaudio
(torchgpu) $ deactivate

Verification:

$ source ~/venvs/torchgpu/bin/activate
(torchgpu) $ python
>>> import torch
>>> torch.cuda.is_available()
>>> exit()
(torchgpu) $ deactivate

Deep Learning Environment (FastAI-CPU)

$ python3 -m venv ~/venvs/fastaicpu
$ source ~/venvs/fastaicpu/bin/activate
(fastaicpu) $ pip install --upgrade pip setuptools wheel
(fastaicpu) $ pip install --upgrade opencv-python opencv-contrib-python
(fastaicpu) $ pip install --upgrade torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu
(fastaicpu) $ pip install --upgrade numpy scipy matplotlib ipython jupyter pandas sympy nose
(fastaicpu) $ pip install --upgrade scikit-learn scikit-image
(fastaicpu) $ pip install --upgrade fastai fastbook
(fastaicpu) $ deactivate

Deep Learning Environment (FastAI-GPU)

$ python3 -m venv ~/venvs/fastaigpu
$ source ~/venvs/fastaigpu/bin/activate
(fastaigpu) $ pip install --upgrade pip setuptools wheel
(fastaigpu) $ pip install --upgrade opencv-python opencv-contrib-python
(fastaigpu) $ pip install --upgrade torch torchvision torchaudio
(fastaigpu) $ pip install --upgrade numpy scipy matplotlib ipython jupyter pandas sympy nose
(fastaigpu) $ pip install --upgrade scikit-learn scikit-image
(fastaigpu) $ pip install --upgrade fastai fastbook
(fastaigpu) $ deactivate

ONNX Runtime Environment

$ python3 -m venv ~/venvs/onnx
$ source ~/venvs/onnx/bin/activate
(onnx) $ pip install --upgrade pip setuptools wheel
(onnx) $ pip install --upgrade onnx onnxruntime-gpu
(onnx) $ deactivate

Verification:

$ source ~/venvs/onnx/bin/activate
(onnx) $ python
>>> import onnxruntime as ort
>>> ort.get_device()
>>> exit()
(onnx) $ deactivate

QT Environment (PySide)

$ python3 -m venv ~/venvs/qt
$ source ~/venvs/qt/bin/activate
(qt) $ pip install --upgrade pip setuptools wheel
(qt) $ pip install --upgrade PySide6
(qt) $ deactivate

Install Miniconda

Install:

$ wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
$ chmod +x Miniconda3-latest-Linux-x86_64.sh

$ ./Miniconda3-latest-Linux-x86_64.sh

$ # Do you wish the installer to initialize Miniconda3
  # by running conda init?
  # yes

$ source ~/miniconda3/bin/activate 
$ conda config --set auto_activate_base false
$ conda deactivate

Activate and Deactivate:

$ conda activate
(base) $ conda deactivate

Managing conda:

(base) $ conda info
(base) $ conda update conda

Managing environments:

(base) $ conda info --envs

(base) $ conda create --name snakes python=3.5
(base) $ conda info --envs

(base) $ conda activate snakes

(snakes) $ python --version

(snakes) $ conda search beautifulsoup4

(snakes) $ conda install beautifulsoup4
(snakes) $ conda list

(snakes) $ conda update beautifulsoup4

(snakes) $ conda uninstall beautifulsoup4
(snakes) $ conda list

(snakes) $ conda deactivate

(base) $ conda remove --name snakes --all

(base) $ conda info --envs

Additional Useful Python Packages


Install Visual Studio Code

Install Python extension for Visual Studio Code:

Recommended Extensions:


Install PyCharm

Enable GPU support for PyCharm Projects:

  • Edit Configurations...
  • Environment variables:
    • PATH=$PATH:/usr/local/cuda-11.8/bin
    • LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64:/usr/local/cuda-11.8/extras/CUPTI/lib64

Install Qt


Install Docker Engine & Docker Compose

Nvidia Container Toolkit:

Make sure you have installed the NVIDIA driver and Docker engine for your Linux distribution Note that you do not need to install the CUDA Toolkit on the host system, but the NVIDIA driver needs to be installed.

$ distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
      && curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
      && curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | \
            sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
            sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list

$ sudo apt update

$ sudo apt-get install --yes nvidia-container-toolkit

$ sudo nvidia-ctk runtime configure --runtime=docker

$ sudo systemctl restart docker

# NOTE: --runtime=nvidia
$ docker container run --rm --runtime=nvidia nvidia/cuda:11.8.0-base-ubuntu22.04 nvidia-smi
# OR
$ docker container run --rm --gpus all nvidia/cuda:11.8.0-base-ubuntu22.04 nvidia-smi

TensorFlow Docker:

# CPU
$ docker run -it tensorflow/tensorflow:latest bash

# GPU
$ docker run --gpus all -it tensorflow/tensorflow:latest-gpu bash

PyTorch Docker:

$ docker run --gpus all --rm -ti --ipc=host pytorch/pytorch:latest

Running ARM Docker Containers:

$ sudo apt install qemu qemu-user-static binfmt-support
$ # Host Architecture: x86_64
$ uname -m
$ # ARM Container Architecture: armv7l
$ docker run --rm arm32v7/debian uname -m

Install Additional Tools

$ sudo apt install ubuntu-restricted-extras
$ sudo apt install virtualbox virtualbox-dkms virtualbox-ext-pack virtualbox-guest-additions-iso
$ sudo apt install curl wget uget tar zip unzip rar unrar
$ sudo apt install gimp vlc ffmpeg
$ sudo apt install kdiff3
  • Ubuntu Software > System Load Indicator

Install Additional Fonts

  • $ mkdir ~/.fonts
  • Copy fonts to ~/.fonts
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment