name: mdlm
channels:
  - pytorch
  - conda-forge
  - defaults
dependencies:
  - python=3.9
  - pytorch=2.1.0
  - torchvision
  - torchaudio
  - pytorch-cuda=11.8
  - cudatoolkit=11.8
  - pip
  - pip:
    - fair-esm
    - transformers
    - datasets
    - accelerate
    - wandb
    - tqdm
    - numpy
    - scipy
    - scikit-learn
    - matplotlib
    - seaborn
    - pandas
    - biopython
    - h5py
    - tensorboard
    - jupyter
    - ipykernel