requirements-cu11.txt

-f https://download.pytorch.org/whl/cu111/torch_stable.html
corpusit==0.1.3
faster_fifo<=1.2.0
nltk<=3.7
numpy==1.20
scipy==1.7.1
numba==0.54
pandas==1.3.5
scikit_learn<=1.1.2
torch==1.10.1+cu111
tqdm<=4.64.1
rich<=12.6.0
typing_extensions==4.4.0
matplotlib==3.5.3
wandb==0.13.4