# Duo's torch 2.3.1 pin predates the project's CUDA 12.4 target.  This uses a
# coherent CUDA 12.4 PyTorch pair; create_all.sh retains Duo's FA 2.7.4.post1.
name: dlb-duo
channels: [pytorch, nvidia, conda-forge]
dependencies:
  - python=3.12
  - 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.15.0
      - einops==0.7.0
      - fsspec==2023.10.0
      - h5py==3.10.0
      - hydra-core==1.3.2
      - lightning==2.2.1
      - nvitop==1.3.2
      - omegaconf==2.3.0
      - pandas==2.2.1
      - pydantic==2.12.4
      - pyyaml==6.0.2
      - rich==13.7.1
      - scikit-learn==1.4.0
      - timm==0.9.16
      - torchmetrics==1.6.1
      - tqdm==4.66.2
      - transformers==4.38.2
      - wandb==0.18.7