fix(ray): fix "GPU" tag deployment

instill-ai · Dec 19, 2024 · 29f83af · 29f83af
1 parent 6abcfd2
commit 29f83af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/ray/ray.go b/pkg/ray/ray.go
@@ -316,7 +316,7 @@ func (r *ray) setHardwareRunOptions(hardware string, numOfGPU string) []string {
 	case SupportedAcceleratorType["GPU"]:
 		runOptions = append(runOptions,
 			fmt.Sprintf("-e %s=%v", EnvTotalVRAM, config.Config.RayServer.Vram),
-			fmt.Sprintf("-e %s=%v", EnvNumOfGPUs, numOfGPU),
+			fmt.Sprintf("-e %s=%v", EnvNumOfGPUs, 1),
 			"--device nvidia.com/gpu=all",
 		)
 	default: