refactored code to three interfaces

Author: JunAr7112

deployments/container/Dockerfile.ubi9

@@ -26,7 +26,7 @@ RUN set -eux; \
         aarch64) ARCH='arm64' ;; \
         *) echo "unsupported architecture" ; exit 1 ;; \
     esac; \
-    wget -nv -O - https://storage.googleapis.com/golang/go${GOLANG_VERSION}.linux-${ARCH}.tar.gz \
+    wget -nv -O - https://dl.google.com/go/go${GOLANG_VERSION}.linux-${ARCH}.tar.gz \
     | tar -C /usr/local -xz
 
 ENV GOPATH /go
@@ -46,6 +46,8 @@ FROM nvcr.io/nvidia/cloud-native/k8s-mig-manager:v0.12.3-ubi9 as mig-manager
 FROM nvcr.io/nvidia/distroless/go:v3.1.13-dev
 
 ENV NVIDIA_VISIBLE_DEVICES=void
+# Preload NVIDIA NVML library from the driver mounted at /driver-root
+ENV LD_PRELOAD=/driver-root/usr/lib/x86_64-linux-gnu/libnvidia-ml.so.1
 
 COPY --from=build /artifacts/nvidia-vgpu-dm /usr/bin/nvidia-vgpu-dm
 COPY --from=build /artifacts/nvidia-k8s-vgpu-dm /usr/bin/nvidia-k8s-vgpu-dm