From ce29375fd09f244c4caf6d986b1cace1b237e292 Mon Sep 17 00:00:00 2001 From: Karl Mutch Date: Fri, 8 Mar 2019 22:14:21 -0800 Subject: [PATCH] FIx issues related to CUDA 9.0 support within Kubernetes and release --- Dockerfile_base | 2 +- cmd/runner/Dockerfile | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile_base b/Dockerfile_base index 53030149d..08d26f5ff 100644 --- a/Dockerfile_base +++ b/Dockerfile_base @@ -94,7 +94,7 @@ ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64 # nvidia-container-runtime ENV NVIDIA_VISIBLE_DEVICES all ENV NVIDIA_DRIVER_CAPABILITIES compute,utility -ENV NVIDIA_REQUIRE_CUDA "cuda>=8.0" +ENV NVIDIA_REQUIRE_CUDA "cuda>=9.0" # Pick up some TF dependencies RUN apt-get update && apt-get install -y --no-install-recommends \ diff --git a/cmd/runner/Dockerfile b/cmd/runner/Dockerfile index d89d0aa8e..adeeeabc3 100644 --- a/cmd/runner/Dockerfile +++ b/cmd/runner/Dockerfile @@ -49,7 +49,7 @@ ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64 # nvidia-container-runtime ENV NVIDIA_VISIBLE_DEVICES all ENV NVIDIA_DRIVER_CAPABILITIES compute,utility -ENV NVIDIA_REQUIRE_CUDA "cuda>=8.0" +ENV NVIDIA_REQUIRE_CUDA "cuda>=9.0" # Pick up some TF dependencies RUN apt-get install -y --no-install-recommends \