From 1cfb25e5d61c69bcbcb07ea0e0baba6d480f9bfc Mon Sep 17 00:00:00 2001 From: Jason Dove Date: Sat, 31 Dec 2022 12:59:06 -0600 Subject: [PATCH] set nvidia capabilities env var --- nvidia.Dockerfile | 2 ++ 1 file changed, 2 insertions(+) diff --git a/nvidia.Dockerfile b/nvidia.Dockerfile index 49508ce..72d6414 100644 --- a/nvidia.Dockerfile +++ b/nvidia.Dockerfile @@ -510,6 +510,8 @@ RUN ldconfig && \ FROM mcr.microsoft.com/dotnet/aspnet:7.0-jammy-amd64 AS dotnet-runtime FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04 as runtime-base +ENV NVIDIA_DRIVER_CAPABILITIES all + RUN apt-get -yqq update && \ apt-get install -yq --no-install-recommends ca-certificates expat libgomp1 libxcb-shape0 libv4l-0 && \ apt-get autoremove -y && \