baby steps 5: pytorch before-and-after

StanfordAHA · Sep 26, 2023 · dc2f2b3 · dc2f2b3
1 parent 2fb88da
commit dc2f2b3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -131,7 +131,9 @@ 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 && \
-  rm -rf $TMPDIR
+  echo -n "BEFORE CLEANUP: " && du -hs /aha && \
+  /bin/rm -rf $TMPDIR && \
+  echo -n "AFTER  CLEANUP: " && du -hs /aha
 ENV TMPDIR=$TMPTMPDIR
 
 WORKDIR /aha