1. Install necessary libs and drivers

sudo apt install nvidia-driver-535

wget https://developer.download.nvidia.com/compute/cudnn/9.2.0/local_installers/cudnn-local-repo-ubuntu2204-9.2.0_1.0-1_amd64.deb
sudo dpkg -i cudnn-local-repo-ubuntu2204-9.2.0_1.0-1_amd64.deb

sudo cp /var/cudnn-local-repo-ubuntu2204-9.2.0/cudnn-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cudnn nvidia-cuda-toolkit -y
nvcc --version

2. Install dependencies

    python -m pip install --upgrade pip
    pip install --upgrade huggingface_hub
    pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
    pip install ultralytics albumentations