chore(cli): use preferred nvidia driver meta pkg name (#2051)

This commit is contained in:
dkeven 2025-11-11 21:13:05 +08:00 committed by GitHub
parent 4cee006a1e
commit 58cdd7de69
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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 {