ipython==8.12.3
matplotlib==3.7.5
numpy==1.24.1
scikit_learn==1.3.2
--extra-index-url https://download.pytorch.org/whl/cu118
torch==2.2.0+cu118
torchmetrics==1.4.0.post0
torchvision==0.17.0+cu118