diff --git a/install/docker/Dockerfile.nvidia b/install/docker/Dockerfile.nvidia index 67c81493f..6cb9265e7 100644 --- a/install/docker/Dockerfile.nvidia +++ b/install/docker/Dockerfile.nvidia @@ -1,6 +1,9 @@ ARG BASE="ghcr.io/koush/scrypted-common:20-jammy-full" FROM $BASE +ENV NVIDIA_DRIVER_CAPABILITIES=all +ENV NVIDIA_VISIBLE_DEVICES=all + # nvidia cudnn/libcublas etc. # for some reason this is not provided by the nvidia container toolkit RUN curl https://raw.githubusercontent.com/koush/scrypted/main/install/docker/install-nvidia-graphics.sh | bash