requirements.txt

# tested with python 3.11.10

torch==2.3.1 --index-url https://download.pytorch.org/whl/cpu
torchvision==0.18.1 --index-url https://download.pytorch.org/whl/cpu
torchaudio==2.3.1 --index-url https://download.pytorch.org/whl/cpu
cuda-python==12.1.0
lightning==2.3.3
torchmetrics==1.4.2
tensorboard==2.18.0
matplotlib==3.9.2
nltk==3.9.1
torchtext==0.18.0
gradio==4.44.1