Merge pull request #1 from hasty-ai/vgpu-support

Support vgpus in autoscaler
cloudfactory · Feb 16, 2022 · b544dd0 · b544dd0
2 parents da4baf7 + bbd7924
commit b544dd0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cluster-autoscaler/utils/gpu/gpu.go b/cluster-autoscaler/utils/gpu/gpu.go
@@ -24,7 +24,7 @@ import (
 
 const (
 	// ResourceNvidiaGPU is the name of the Nvidia GPU resource.
-	ResourceNvidiaGPU = "nvidia.com/gpu"
+	ResourceNvidiaGPU = "k8s.amazonaws.com/vgpu"
 	// DefaultGPUType is the type of GPU used in NAP if the user
 	// don't specify what type of GPU his pod wants.
 	DefaultGPUType = "nvidia-tesla-k80"