set nvidia capabilities env var

2022-12-31 12:59:06 -06:00
parent 1f3e7b362a
commit 1cfb25e5d6
1 changed files with 2 additions and 0 deletions
--- 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 && \