# 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