From 6e841df14d631b7e157217567fe7b54b7269c106 Mon Sep 17 00:00:00 2001 From: romi1502 Date: Fri, 19 Jun 2020 17:23:32 +0200 Subject: [PATCH] set back NVIDIA_VISIBLE_DEVICES --- docker/cuda-10-0.dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/docker/cuda-10-0.dockerfile b/docker/cuda-10-0.dockerfile index a566eb55..50af97c0 100644 --- a/docker/cuda-10-0.dockerfile +++ b/docker/cuda-10-0.dockerfile @@ -5,7 +5,8 @@ ENV CUDA_VERSION 10.0.130 ENV CUDA_PKG_VERSION 10-0=$CUDA_VERSION-1 ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH} ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64 -# Note: differ from original image, picked from old dzr_audio spec. + +ENV NVIDIA_VISIBLE_DEVICES=all ENV NVIDIA_DRIVER_CAPABILITIES compute,utility ENV NVIDIA_REQUIRE_CUDA "cuda>=10.0 brand=tesla,driver>=384,driver<385 brand=tesla,driver>=410,driver<411" ENV NCCL_VERSION 2.4.2