From 0183be7b1018cc50f337ed33fb9e948436e6f311 Mon Sep 17 00:00:00 2001
From: josemduarte <jose.m.duarte@gmail.com>
Date: Fri, 30 Aug 2024 10:21:38 -0700
Subject: [PATCH] Adding deepspeed dep to Dockerfile

---
 Dockerfile | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/Dockerfile b/Dockerfile
index 83530031..3686806e 100644
--- 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
\ No newline at end of file