chore(helm): fix missing nvidia cdi spec for helm deployment

instill-ai · Mar 19, 2024 · 200b245 · 200b245
1 parent cb72cd8
commit 200b245
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/charts/core/templates/ray-service/ray-service.yaml b/charts/core/templates/ray-service/ray-service.yaml
@@ -169,7 +169,14 @@ spec:
               lifecycle:
                 postStart:
                   exec:
-                    command: ["/bin/sh","-c","/home/ray/script/start_ray_serve.sh"]
+                    command:
+                      - "/bin/bash"
+                      - "-c"
+                      - >
+                        if [[ -n $${NVIDIA_VISIBLE_DEVICES} ]]; then
+                          sudo nvidia-ctk cdi generate --output=/etc/cdi/nvidia.yaml;
+                        fi;
+                        /home/ray/script/start_ray_serve.sh
                 preStop:
                   exec:
                     command: ["/bin/sh","-c","ray stop"]