torch
torchvision
cuda-python==11.8
nvidia-pyindex
nvidia-cudnn
pandas
tiktoken
tqdm