Adding deepspeed dep to Dockerfile

josemduarte · Aug 30, 2024 · 0183be7 · 0183be7
1 parent 6f63267
commit 0183be7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -33,3 +33,7 @@ RUN wget -q -P /opt/openfold/openfold/resources \
     https://git.scicore.unibas.ch/schwede/openstructure/-/raw/7102c63615b64735c4941278d92b554ec94415f8/modules/mol/alg/src/stereo_chemical_props.txt
 WORKDIR /opt/openfold
 RUN python3 setup.py install
+
+# for deepspeed training/inference
+RUN git clone https://github.com/NVIDIA/cutlass.git /opt/cutlass
+ENV CUTLASS_PATH /opt/cutlass