scrubbed torch layer

StanfordAHA · Sep 25, 2023 · fae8d1f · fae8d1f
1 parent 3258626
commit fae8d1f
Showing 1 changed file with 18 additions and 6 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -150,18 +150,30 @@ RUN printf -- "--- TIME %s %s\n" `date +%H:%M` sam && \
     make sam && \
     source /aha/bin/activate && pip install scipy numpy pytest && pip install -e .
 
-# Install torch (need big tmp folder)
-WORKDIR /aha
-RUN mkdir -p /aha/tmp/torch_install/
 # Save (and later restore) existing value for TMPDIR, if any
 ENV TMPTMPDIR=$TMPDIR
 ENV TMPDIR=/aha/tmp/torch_install/
-RUN source /aha/bin/activate && pip install --cache-dir=$TMPDIR --build=$TMPDIR torch==1.7.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
-RUN rm -rf $TMPDIR
+
+# Install torch (need big tmp folder)
+WORKDIR /aha
+RUN printf -- "--- TIME %s %s\n" `date +%H:%M` torch && \
+    mkdir -p /aha/tmp/torch_install/ && \
+    source /aha/bin/activate && \
+    pip install --cache-dir=$TMPDIR --build=$TMPDIR torch==1.7.1+cpu -f https://download.pytorch.org/whl/torch_stable.html && \
+    /bin/rm -rf $TMPDIR
+
+# Restore original value of TMPDIR
 ENV TMPDIR=$TMPTMPDIR
 
 WORKDIR /aha
-RUN source bin/activate && pip install urllib3==1.26.15 && pip install wheel six && pip install systemrdl-compiler peakrdl-html && pip install -e . && pip install packaging==21.3 && pip install -e ./pono/deps/smt-switch/build/python && pip install -e pono/build/python/ && aha deps install
+RUN source bin/activate && \
+  pip install urllib3==1.26.15 && \
+  pip install wheel six && \
+  pip install systemrdl-compiler peakrdl-html && \
+  pip install -e . && \
+  pip install packaging==21.3 && \
+  pip install -e ./pono/deps/smt-switch/build/python && \
+  pip install -e pono/build/python/ && aha deps install
 
 WORKDIR /aha