diff --git a/pkg/scheduler/api/node_info.go b/pkg/scheduler/api/node_info.go
index c66accdb164..9cf18ad0ea6 100644
--- a/pkg/scheduler/api/node_info.go
+++ b/pkg/scheduler/api/node_info.go
@@ -153,17 +153,19 @@ func (ni *NodeInfo) setNodeGPUInfo(node *v1.Node) {
 	if node == nil {
 		return
 	}
-	if memory, ok := node.Status.Capacity[VolcanoGPUNumber]; ok {
-		ni.GPUTotalMemory = memory.Value()
+	memory, ok := node.Status.Capacity[VolcanoGPUResource]
+	if !ok {
+		return
 	}
+	ni.GPUTotalMemory = memory.Value()
 
 	gpuNumber, ok := node.Status.Capacity[VolcanoGPUNumber]
-	if ok {
-		ni.GPUNumber = gpuNumber.Value()
+	if !ok {
+		return
 	}
+	ni.GPUNumber = gpuNumber.Value()
 
 	memoryPerCard := uint(ni.GPUTotalMemory / ni.GPUNumber)
-
 	for i := 0; i < int(ni.GPUNumber); i++ {
 		ni.GPUDevices[i] = NewGPUDevice(i, memoryPerCard)
 	}