1. Install necessary libs and drivers

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

Windows

• Install CUDA

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
  pip install -r requirements.txt
  
  pip uninstall -y torch
  pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126
  

3. Fix possible problems

• cv2.error: OpenCV(4.10.0) ...\window.cpp:1301: error: (-2:Unspecified error)

  pip uninstall -y opencv-python 
  pip install opencv-python
  

• fbgemm.dll error (Windows specific)

  copypaste tests\libomp140.x86_64.dll to C:\Windows\System32