| # Linux/NVIDIA CUDA 12.4 server environment. FlashAttention is installed by | |
| # create_all.sh after PyTorch so its build can discover the selected torch ABI. | |
| name: dlb-flm | |
| channels: [pytorch, nvidia, conda-forge] | |
| dependencies: | |
| - python=3.11 | |
| - pip=24.2 | |
| - pytorch=2.5.1 | |
| - mkl<2024.1 | |
| - pytorch-cuda=12.4 | |
| - torchvision=0.20.1 | |
| - cuda-toolkit=12.4 | |
| - pip: | |
| - datasets==3.6.0 | |
| - einops==0.8.1 | |
| - entmax==1.3 | |
| - fsspec==2025.3.0 | |
| - huggingface_hub==0.36.0 | |
| - hf_transfer==0.1.9 | |
| - hydra-core==1.3.2 | |
| - lightning==2.6.0 | |
| - numpy==2.3.2 | |
| - omegaconf==2.3.0 | |
| - pydantic==2.12.4 | |
| - pyyaml==6.0.2 | |
| - rich==14.2.0 | |
| - scipy==1.16.3 | |
| - timm==1.0.22 | |
| - tokenizers==0.22.1 | |
| - torchmetrics==1.8.2 | |
| - transformers==4.57.3 | |
| - wandb==0.23.1 | |