torch
torchvision
cuda-python==11.8
#nvidia-pyindex
#nvidia-tensorflow[horovod]
pandas
tiktoken
tqdm
pyyaml
datasets
transformers
wandb