@@ -469,7 +469,7 @@ func tkeGetInstanceAdvancedPara(dMap map[string]interface{}, meta interface{}) (
469469 }
470470
471471 // get gpu_args
472- if v , ok := dMap ["gpu_args" ]; ok {
472+ if v , ok := dMap ["gpu_args" ]; ok && len ( v .([] interface {})) > 0 {
473473 gpuArgs := v .([]interface {})[0 ].(map [string ]interface {})
474474
475475 var (
@@ -481,27 +481,27 @@ func tkeGetInstanceAdvancedPara(dMap map[string]interface{}, meta interface{}) (
481481 )
482482 tkeGpuArgs := tke.GPUArgs {}
483483 tkeGpuArgs .MIGEnable = & migEnable
484- if driver != nil && len (driver ) > 0 {
484+ if len (driver ) > 0 {
485485 tkeGpuArgs .Driver = & tke.DriverVersion {
486486 Version : helper .String (driver ["version" ].(string )),
487487 Name : helper .String (driver ["name" ].(string )),
488488 }
489489 }
490- if cuda != nil && len (cuda ) > 0 {
490+ if len (cuda ) > 0 {
491491 tkeGpuArgs .CUDA = & tke.DriverVersion {
492492 Version : helper .String (cuda ["version" ].(string )),
493493 Name : helper .String (cuda ["name" ].(string )),
494494 }
495495 }
496- if cudnn != nil && len (cudnn ) > 0 {
496+ if len (cudnn ) > 0 {
497497 tkeGpuArgs .CUDNN = & tke.CUDNN {
498498 Version : helper .String (cudnn ["version" ].(string )),
499499 Name : helper .String (cudnn ["name" ].(string )),
500500 DocName : helper .String (cudnn ["doc_name" ].(string )),
501501 DevName : helper .String (cudnn ["dev_name" ].(string )),
502502 }
503503 }
504- if customDriver != nil && len (customDriver ) > 0 {
504+ if len (customDriver ) > 0 {
505505 tkeGpuArgs .CustomDriver = & tke.CustomDriver {
506506 Address : helper .String (customDriver ["address" ].(string )),
507507 }
0 commit comments