Dockerfile with minimal env needed for training

f86fc334 · Marta · e69b2a11 · f86fc334 · f86fc334
Commit f86fc334 authored Oct 29, 2021 by Marta
Show whitespace changes
Inline Side-by-side

Showing with 19 additions and 0 deletions

Dockerfile Dockerfile +11 -0

requirements_minimal.txt requirements_minimal.txt +8 -0

No files found.
--- a/Dockerfile
+++ b/Dockerfile
+ARG IMAGE_NAME=nvcr.io/nvidia/pytorch:21.06-py3
+FROM ${IMAGE_NAME} AS base
+ENV DEBIAN_FRONTEND=noninteractive
+COPY . /workspace/openfold
+WORKDIR /workspace/openfold
+RUN pip install -r requirements_minimal.txt
+RUN python setup.py install
--- a/requirements_minimal.txt
+++ b/requirements_minimal.txt
+biopython==1.79
+deepspeed==0.5.3
+dm-tree==0.1.6
+ml-collections==0.1.0
+ninja==1.10.2
+tensorboardX==1.8
+triton==1.0.0
+git+https://github.com/PyTorchLightning/pytorch-lightning.git@6d79184ec50d9f80448013cc2c01b5b744a70b44