Update Dockerfile

In-For-Disaster-Analytics · Feb 29, 2024 · 313d3a1 · 313d3a1
1 parent fb1254c
commit 313d3a1
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.3.0-devel-ubuntu22.04
+FROM nvidia/cuda:12.3.0-runtime-ubuntu22.04
 
 LABEL maintainer="TACC-ACI-WMA <[email protected]>"
 
@@ -32,8 +32,7 @@ RUN pip install --upgrade --no-cache-dir \
     wheel
 
 # Install jupyterlab and ML packages using host cache
-RUN --mount=type=cache,target=/root/.cache/pip \
-    pip install \
+RUN pip install --no-cache-dir\
     jupyterlab \
     tensorflow[and-cuda] \
     torch