baby steps 1: remove pytorch tmpdir

StanfordAHA · Sep 26, 2023 · a902555 · a902555
1 parent 5ecdb0a
commit a902555
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -119,8 +119,9 @@ 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
+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
 ENV TMPDIR=$TMPTMPDIR
 
 WORKDIR /aha