diff --git a/sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/files/bin/setup-nvidia b/sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/files/bin/setup-nvidia
index c694e40f03..38c5186056 100644
--- a/sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/files/bin/setup-nvidia
+++ b/sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/files/bin/setup-nvidia
@@ -131,6 +131,8 @@ function install_and_load() {
     mknod -m 666 /dev/nvidia-uvm c $D 0
   fi
 
+  nvidia-ctk cdi generate --output=/etc/cdi/nvidia.yaml
+
   if [ "${NVIDIA_PERSISTENCE_MODE}" = "on" ]; then
     nvidia-persistenced
   fi