set back NVIDIA_VISIBLE_DEVICES

Eugene-Y · Jun 19, 2020 · 6e841df · 6e841df
1 parent 331ab28
commit 6e841df
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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