File size: 827 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 | # 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
|