diff --git a/supervisely/train/config.json b/supervisely/train/config.json
index b54c897098da..7425e7451805 100644
--- a/supervisely/train/config.json
+++ b/supervisely/train/config.json
@@ -15,7 +15,8 @@
   "main_script": "supervisely/train/src/sly_train.py",
   "gui_template": "supervisely/train/src/gui.html",
   "task_location": "workspace_tasks",
-  "gpu": "required",
+  "need_gpu": true,
+  "gpu": "preferred",
   "icon": "https://i.imgur.com/HZCM0QM.png",
   "icon_cover": true,
   "context_menu": {
@@ -30,4 +31,4 @@
   "license": {
     "type": "AGPL-3.0"
   }
-}
\ No newline at end of file
+}