Merge pull request #1781 from StanfordAHA/torch_remove_tmp_files

Update Dockerfile - remove tmp install for torch

Dockerfile

@@ -116,8 +116,12 @@ RUN 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
+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