File size: 839 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 | # CANDI's exact torch 2.3.0 pin does not provide a CUDA 12.4 Conda pairing;
# the CUDA 12.4 PyTorch baseline is used and its FA 2.6.1 build stays separate.
# SciPy 1.16 requires newer NumPy, so the compatible stable 1.11.4 is used.
name: dlb-candi
channels: [pytorch, nvidia, conda-forge]
dependencies:
- python=3.11
- pip=24.2
- pytorch=2.5.1
- mkl<2024.1
- pytorch-cuda=12.4
- cuda-toolkit=12.4
- pip:
- datasets==2.19.0
- einops==0.8.1
- evaluate==0.4.2
- fsspec==2024.2.0
- huggingface_hub==0.34.4
- hydra-core==1.3.2
- lightning==2.5.5
- numpy==1.24.3
- omegaconf==2.3.0
- pydantic==2.12.4
- pyyaml==6.0.2
- rich==14.2.0
- scipy==1.11.4
- tokenizers==0.22.1
- torchmetrics==1.8.2
- tqdm==4.66.4
- transformers==4.56.1
|