ci: update Dockerfile.common with nvidia env

Kitware · May 26, 2024 · f98df78 · f98df78
1 parent b2c095e
commit f98df78
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/docker/Dockerfile.common b/docker/Dockerfile.common
@@ -4,6 +4,9 @@ FROM ${BASE_IMAGE}
 # Necessary to install tzdata. It will default to UTC.
 ENV DEBIAN_FRONTEND=noninteractive
 
+# Make sure we enable all GPU if we have one
+ENV NVIDIA_DRIVER_CAPABILITIES=all
+
 RUN apt-get update && \
     apt-get install -y \
       wget \