FIx issues related to CUDA 9.0 support within Kubernetes and release

leaf-ai · Mar 9, 2019 · ce29375 · ce29375
1 parent 4a8c474
commit ce29375
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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 \