diff --git a/deployments/container/Dockerfile.ubi8 b/deployments/container/Dockerfile.ubi8 index 805345ed2..7721e8265 100644 --- a/deployments/container/Dockerfile.ubi8 +++ b/deployments/container/Dockerfile.ubi8 @@ -26,7 +26,7 @@ RUN export CGO_LDFLAGS_ALLOW='-Wl,--unresolved-symbols=ignore-in-object-files' & go build -ldflags="-s -w -X 'main.Version=${VERSION}'" -v ./cmd/nvidia-device-plugin -FROM nvcr.io/nvidia/${CUDA_IMAGE}:${CUDA_VERSION}-base-${BASE_DIST} +FROM nvidia/${CUDA_IMAGE}:${CUDA_VERSION}-base-${BASE_DIST} ENV NVIDIA_DISABLE_REQUIRE="true" ENV NVIDIA_VISIBLE_DEVICES=all diff --git a/deployments/container/Dockerfile.ubuntu b/deployments/container/Dockerfile.ubuntu index 493a05bd1..e4e550488 100644 --- a/deployments/container/Dockerfile.ubuntu +++ b/deployments/container/Dockerfile.ubuntu @@ -26,7 +26,7 @@ RUN export CGO_LDFLAGS_ALLOW='-Wl,--unresolved-symbols=ignore-in-object-files' & go build -ldflags="-s -w -X 'main.Version=${VERSION}'" -v ./cmd/nvidia-device-plugin -FROM nvcr.io/nvidia/${CUDA_IMAGE}:${CUDA_VERSION}-base-${BASE_DIST} +FROM nvidia/${CUDA_IMAGE}:${CUDA_VERSION}-base-${BASE_DIST} ENV NVIDIA_DISABLE_REQUIRE="true" ENV NVIDIA_VISIBLE_DEVICES=all