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