fix dependency conflicts for ubuntu 22 tensorflow images

ika-rwth-aachen · Dec 9, 2024 · d99d549 · d99d549
1 parent dc61965
commit d99d549
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -250,6 +250,7 @@ RUN if [[ -n $TORCH_VERSION ]]; then \
 ARG TF_VERSION
 RUN if [[ -n $TF_VERSION ]]; then \
         if [[ "$TARGETARCH" == "amd64" ]]; then \
+            if [[ $UBUNTU_VERSION == "22.04" ]]; then pip install opencv-python matplotlib; fi && \
             pip3 install tensorflow==${TF_VERSION}; \
         elif [[ "$TARGETARCH" == "arm64" ]]; then \
             apt-get update && \