diff --git a/cli/pkg/gpu/tasks.go b/cli/pkg/gpu/tasks.go
index 1146fac21..9c937fe19 100644
--- a/cli/pkg/gpu/tasks.go
+++ b/cli/pkg/gpu/tasks.go
@@ -131,8 +131,8 @@ func (t *InstallCudaDriver) Execute(runtime connector.Runtime) error {
 		return errors.Wrap(err, "failed to apt-get install nvidia-open")
 	}
 
-	if _, err := runtime.GetRunner().SudoCmd("apt-get -y install nvidia-kernel-open-575", false, true); err != nil {
-		return errors.Wrap(errors.WithStack(err), "Failed to apt-get install nvidia-kernel-open-575")
+	if _, err := runtime.GetRunner().SudoCmd("apt-get -y install nvidia-open-575", false, true); err != nil {
+		return errors.Wrap(errors.WithStack(err), "Failed to apt-get install nvidia-open-575")
 	}
 
 	if t.SkipNVMLCheckAfterInstall {