• Rough video guide

• Test if Cuda is working for Pytorch: python -c "import torch; print(torch.cuda.is_available())"
• Update nvidia drivers
• If Cuda works but not in virtual environment update in the virtual environment:

source /venv/bin/activate
pip install -r requirements.txt
pip install --upgrade pip setuptools wheel
pip install --upgrade torch torchvision

• cheat sheet
• cheat sheet with example images