Commit 13f0f6fe authored by Jennifer Wei's avatar Jennifer Wei
Browse files

Adds libgcc and pytorch::pytorch to environment.yml

parent a5c69a79
name: openfold_venv
name: openfold-venv
channels:
- conda-forge
- bioconda
- pytorch
dependencies:
- python=3.9
- libgcc=7.2
- setuptools=59.5.0
- pip
- openmm=7.7
- pdbfixer
- cudatoolkit==11.3.*
- pytorch=1.12.*
- pytorch-lightning==1.5.10
- biopython==1.79
- numpy==1.21
......@@ -27,8 +28,9 @@ dependencies:
- bioconda::hmmer==3.3.2
- bioconda::hhsuite==3.3.0
- bioconda::kalign2==2.04
- pytorch::pytorch=1.12.*
- pip:
- deepspeed==0.5.10
- dm-tree==0.1.6
- git+https://github.com/NVIDIA/dllogger.git
- git+https://github.com/Dao-AILab/flash-attention.git@v0.2.1
- git+https://github.com/Dao-AILab/flash-attention.git@5b838a8
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment