diff --git a/tools/cli/commands/creategpu.py b/tools/cli/commands/creategpu.py
index d3e6c3cba..4d8ab369f 100644
--- a/tools/cli/commands/creategpu.py
+++ b/tools/cli/commands/creategpu.py
@@ -66,7 +66,6 @@
     User=root
     Type=oneshot
     RemainAfterExit=true
-    ExecStartPre=docker-credential-gcr configure-docker
     ExecStart=/bin/bash -c 'while [ ! -e /mnt/disks/datalab-pd/tmp ]; do \
         sleep 1; \
         done'
@@ -75,7 +74,7 @@
   permissions: 0755
   owner: root
   content: |
-    NVIDIA_DRIVER_VERSION=390.46
+    NVIDIA_DRIVER_VERSION=418.67
     COS_NVIDIA_INSTALLER_CONTAINER=gcr.io/cos-cloud/cos-gpu-installer:latest
     NVIDIA_INSTALL_DIR_HOST=/var/lib/nvidia
     NVIDIA_INSTALL_DIR_CONTAINER=/usr/local/nvidia
@@ -94,6 +93,7 @@
 
     [Service]
     User=root
+    Environment="HOME=/home/datalab"
     Type=oneshot
     RemainAfterExit=true
     EnvironmentFile=/etc/nvidia-installer-env