**1. Install necessary libs and drivers**

https://www.python.org/downloads/

Windows
   * [Install CUDA](https://developer.nvidia.com/cuda-12-1-0-download-archive)

Linux
* ```
  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**

  * (Optional) Install virtual env:
    ``` 
    pip install virtualenv
    python -m venv env
    env\Scripts\activate
    ```

```
    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 git+https://github.com/airockchip/ultralytics_yolov8.git
    pip install albumentations
```

**3. Fix possible problems**
  * cv2.error: OpenCV(4.10.0) ...\window.cpp:1301: error: (-2:Unspecified error)
      ```
      pip uninstall opencv-python 
      pip install opencv-python
      ```
  * fbgemm.dll  error (Windows specific)
      ```
      copypaste libomp140.x86_64.dll to C:\Windows\System32
      ```