# Check the available (prebuilt) PyTorch - CUDA Toolkit packages here:
# https://download.pytorch.org/whl/torch/
#
# The installed version here is the latest PyTorch version compatible with the
# CUDA version 11.3.
# 
# The version of CUDA that can be used depends on the NVIDIA driver version, and
# can be found with the `nvidia-smi` command (e.g. `CUDA Version: 11.4`).
# Beware, this only displays the highest version that can be handled.
--extra-index-url https://download.pytorch.org/whl/cu113
torch==1.12.1+cu113
torchvision==0.13.1+cu113
kornia==0.6.12
Pillow==9.5.0

scikit-learn==1.2.2
matplotlib==3.7.1
pytorch-lightning==2.0.0

mlflow==2.2.2
cffi==1.15.1
tensorboard==2.12.0
torch-tb-profiler==0.4.1

pylint==2.17.2
black==23.3.0
mypy==1.1.1