Update Dockerfile

tensorflow/tensorflow:latest-gpu as base image removed venv again
mindee · Sep 18, 2023 · f80476c · f80476c
1 parent d4f7e3d
commit f80476c
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,20 +1,21 @@
-FROM nvcr.io/nvidia/tensorflow:22.10.1-tf2-py3
+# Use the TensorFlow GPU image as the base image. This image also works with CPU-only setups
+FROM tensorflow/tensorflow@sha256:b4676741c491bff3d0f29c38c369281792c7d5c5bfa2b1aa93e5231a8d236323
 
 ENV PYTHONUNBUFFERED=1 
 ENV PYTHONDONTWRITEBYTECODE=1 
 ENV DOCTR_CACHE_DIR=/app/.cache 
-ENV PATH=/app/venv/bin:$PATH
 
 WORKDIR /app
 
 COPY . .
 
+# Install necessary dependencies for video processing and GUI operations
 RUN apt-get update \
-    && apt-get install --no-install-recommends ffmpeg libsm6 libxext6 python3-venv -y \
+    && apt-get install --no-install-recommends ffmpeg libsm6 libxext6 -y \
     && apt-get autoremove -y \
     && rm -rf /var/lib/apt/lists/* 
 
-RUN python -m venv /app/venv \
-    && pip install --upgrade pip setuptools wheel \
+# Install the current application with TensorFlow extras and modify permissions
+RUN pip install --upgrade pip setuptools wheel \
     && pip install -e .[tf] \
     && chmod -R a+w /app