fix: gpu support os version

beclab · Aug 22, 2024 · 779eb12 · 779eb12
1 parent b5d6daa
commit 779eb12
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/pkg/phase/system/prepare.go b/pkg/phase/system/prepare.go
@@ -8,6 +8,7 @@ import (
 	"bytetrade.io/web3os/installer/pkg/bootstrap/patch"
 	"bytetrade.io/web3os/installer/pkg/bootstrap/precheck"
 	"bytetrade.io/web3os/installer/pkg/common"
+	"bytetrade.io/web3os/installer/pkg/constants"
 	"bytetrade.io/web3os/installer/pkg/container"
 	"bytetrade.io/web3os/installer/pkg/core/module"
 	"bytetrade.io/web3os/installer/pkg/core/pipeline"
@@ -47,9 +48,9 @@ func PrepareSystemPhase(runtime *common.KubeRuntime) *pipeline.Pipeline {
 	}
 
 	m = append(m,
-		&gpu.InstallDepsModule{Skip: !runtime.Arg.GPU.Enable},
-		&gpu.RestartK3sServiceModule{Skip: !runtime.Arg.GPU.Enable},
-		&gpu.RestartContainerdModule{Skip: !runtime.Arg.GPU.Enable},
+		&gpu.InstallDepsModule{Skip: !runtime.Arg.GPU.Enable || !isSupportOs()},
+		&gpu.RestartK3sServiceModule{Skip: !runtime.Arg.GPU.Enable || !isSupportOs()},
+		&gpu.RestartContainerdModule{Skip: !runtime.Arg.GPU.Enable || !isSupportOs()},
 		&gpu.InstallPluginModule{Skip: true},
 		&terminus.PreparedModule{},
 	)
@@ -60,3 +61,11 @@ func PrepareSystemPhase(runtime *common.KubeRuntime) *pipeline.Pipeline {
 		Runtime: runtime,
 	}
 }
+
+func isSupportOs() bool {
+	if constants.OsPlatform == common.Ubuntu && (strings.Contains(constants.OsVersion, "20.") || strings.Contains(constants.OsVersion, "22.")) {
+		return true
+	}
+
+	return false
+}