[Fix] 🐛 Fix GPU Device Error at NoCard Instances #14

services/dispatcher/control.go

@@ -61,7 +61,7 @@ func control(serverID uint, data Data) (err error) {
 			if err != nil {
 				ctx := context.Background()
 				redis.RawDB.IncrBy(ctx, "remain_gpu:server:"+strconv.Itoa(int(serverID)), int64(instance.GpuCount))
-				database.DB.Model(&instance).Update("status", models.InstanceStatusFail).Update("from_action", models.InstanceActionRestart)
+				database.DB.Model(&instance).Update("status", models.InstanceStatusFail).Update("from_action", models.InstanceActionStart)
 				lc.Error("instance restart error: %v", err)
 				return
 			}

services/instanceController/createInstance.go

@@ -48,6 +48,7 @@ func createInstance(ip string, port int, apikey string,
 		},
 		Env: []string{
 			"NVIDIA_DRIVER_CAPABILITIES=video,compute,utility",
+			"NVIDIA_VISIBLE_DEVICES=none",
 		},
 	}
 
@@ -63,7 +64,7 @@ func createInstance(ip string, port int, apikey string,
 	if config.GetSystemMountDir() != "" {
 		data.Binds = append(data.Binds, bindStruct{
 			Src:  config.GetSystemMountDir(),
-			Dest: "/root/megrez",
+			Dest: "/root/megrez-pub",
 		})
 	}