feat(installer): add gpu commands (#826)

* feat: add node selector * feat(installer): install gpu driver & plugin by default * fix: label bug * fix: update installer --------- Co-authored-by: liuyu <>
beclab · Jan 6, 2025 · 96a7579 · 96a7579
1 parent aae7a4c
commit 96a7579
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 2 deletions.
diff --git a/build/installer/deploy/device-plugin.yaml b/build/installer/deploy/device-plugin.yaml
@@ -28,6 +28,8 @@ spec:
     spec:
       runtimeClassName: nvidia # Explicitly request the runtime
       priorityClassName: system-node-critical
+      nodeSelector:
+        gpu.bytetrade.io/cuda-supported: 'true'
       initContainers:
       - name: init-dir
         image: busybox:1.28

diff --git a/build/installer/deploy/nvidia-device-plugin.yml b/build/installer/deploy/nvidia-device-plugin.yml
@@ -44,6 +44,8 @@ spec:
       # be rescheduled after a failure.
       # See https://kubernetes.io/docs/tasks/administer-cluster/guaranteed-scheduling-critical-addon-pods/
       priorityClassName: "system-node-critical"
+      nodeSelector:
+        gpu.bytetrade.io/cuda-supported: 'true'
       containers:
       - image: nvcr.io/nvidia/k8s-device-plugin:v0.16.1
         name: nvidia-device-plugin-ctr

diff --git a/build/installer/deploy/scheduler.yaml b/build/installer/deploy/scheduler.yaml
@@ -28,6 +28,8 @@ spec:
     spec:
       runtimeClassName: nvidia # Explicitly request the runtime
       priorityClassName: system-node-critical
+      nodeSelector:
+        gpu.bytetrade.io/cuda-supported: 'true'
       initContainers:
       - name: init-dir
         image: busybox:1.28

diff --git a/build/installer/install.ps1 b/build/installer/install.ps1
@@ -38,7 +38,7 @@ if (-Not (Test-Path $CLI_PROGRAM_PATH)) {
   New-Item -Path $CLI_PROGRAM_PATH -ItemType Directory
 }
 
-$CLI_VERSION = "0.1.87"
+$CLI_VERSION = "0.1.89"
 $CLI_FILE = "olares-cli-v{0}_windows_{1}.tar.gz" -f $CLI_VERSION, $arch
 $CLI_URL = "{0}/{1}" -f $downloadUrl, $CLI_FILE
 $CLI_PATH = "{0}{1}" -f $CLI_PROGRAM_PATH, $CLI_FILE

diff --git a/build/installer/install.sh b/build/installer/install.sh
@@ -74,7 +74,7 @@ if [ -z ${cdn_url} ]; then
     cdn_url="https://dc3p1870nn3cj.cloudfront.net"
 fi
 
-CLI_VERSION="0.1.87"
+CLI_VERSION="0.1.89"
 CLI_FILE="olares-cli-v${CLI_VERSION}_linux_${ARCH}.tar.gz"
 if [[ x"$os_type" == x"Darwin" ]]; then
     CLI_FILE="olares-cli-v${CLI_VERSION}_darwin_${ARCH}.tar.gz"