diff --git a/rdma-tools/image/install-tools.sh b/rdma-tools/image/install-tools.sh
index 3f7ecab..8b86580 100644
--- a/rdma-tools/image/install-tools.sh
+++ b/rdma-tools/image/install-tools.sh
@@ -45,7 +45,8 @@ apt-get install -y apt-utils
 apt-get install -y ca-certificates
 apt-get update
 # nvidia-smi
-apt-get install -y nvidia-utils-555
+apt-get install -y cuda-drivers-555
+#apt-get install -y nvidia-utils-555
 
 apt-get purge --auto-remove
 apt-get clean