prot_xlstm_env.yml

name: prot_xlstm
channels:
  - pytorch
  - nvidia
  - conda-forge
  - defaults
dependencies:
  - cuda=12.1
  - cuda-nvcc=12.1
  - gxx_linux-64=11.2.0
  - python=3.11
  - pip
  - pytorch=2.2.0
  - pytorch-cuda=12.1
  - cmake
  - ninja
  - pip:
    - accelerate>=0.26.0
    - biopython #==1.83
    - bottleneck #==1.4.2
    - dacite #==1.8.1
    - ipykernel #==6.29.3  
    - mamba_ssm==1.2.0
    - matplotlib #==3.8.4
    - numpy<2.0 #==1.26.4
    - omegaconf #==2.3.0
    - pandas #==2.2.2
    - pyhmmer #==0.10.15
    - rich #==13.7.1
    - scipy #==1.13.0
    - seaborn #==0.13.2
    - torchmetrics #==1.2.1
    - tqdm #==4.66.4
    - transformers==4.44.2
    - tueplots #==0.0.17
    - wandb #==0.17.0