| # 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 | |