torch
torchvision
cuda-python==11.8
nvidia-pyindex
nvidia-tensorflow[horovod]
pandas
tiktoken
tqdm