File size: 1,024 Bytes
784f572 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 | # Replaces upstream's obsolete nightly CPU torchdata command with stable,
# CUDA-compatible torchdata. FlashAttention is built after the torch install.
name: dlb-sdtt
channels: [pytorch, nvidia, conda-forge]
dependencies:
- python=3.10
- pip=24.2
- pytorch=2.5.1
- mkl<2024.1
- pytorch-cuda=12.4
- torchvision=0.20.1
- cuda-toolkit=12.4
- pip:
- datasets==2.18.0
- einops==0.7.0
- evaluate==0.4.2
- fancy-einsum==0.0.3
- fsspec==2024.2.0
- gpustat==1.1.1
- huggingface-hub==0.21.1
- hydra-core==1.3.2
- hydra-submitit-launcher==1.2.0
- lightning==2.2.1
- loguru==0.7.2
- mauve-text==0.3.0
- nltk==3.9.1
- omegaconf==2.3.0
- pandas==2.2.1
- pydantic==2.12.4
- pyyaml==6.0.2
- safetensors==0.4.2
- setuptools==80.9.0
- tensorboard==2.15.2
- timm==0.9.16
- tokenizers==0.15.2
- torchdata==0.8.0
- transformers==4.38.2
- vendi-score==0.0.3
- wandb==0.13.5
|