# RDLM requires Python 3.9 and PyTorch 2.3.1.  That PyTorch release ships the
# CUDA 12.1 runtime; an NVIDIA 12.4 driver remains backward compatible.
name: dlb-rdlm
channels: [pytorch, nvidia, conda-forge]
dependencies:
  - python=3.9
  - pip=24.2
  - pytorch=2.3.1
  - mkl<2024.1
  - pytorch-cuda=12.1
  - pip:
      - accelerate==1.2.1
      - datasets==3.2.0
      - einops==0.8.0
      - eval-type-backport==0.2.2
      - fsspec==2024.9.0
      - hydra-core==1.3.2
      - hydra-joblib-launcher==1.2.0
      - numpy==1.26.4
      - omegaconf==2.3.0
      - pydantic==2.12.4
      - pyyaml==6.0.2
      - scipy==1.13.1
      - tokenizers==0.21.0
      - tqdm==4.67.1
      - transformers==4.47.1
      - wandb==0.19.1