diff --git a/supervisely/train/config.json b/supervisely/train/config.json
index f8f51ffd414f..ae8803b8aceb 100644
--- a/supervisely/train/config.json
+++ b/supervisely/train/config.json
@@ -15,8 +15,8 @@
   "main_script": "supervisely/train/src/sly_train.py",
   "gui_template": "supervisely/train/src/gui.html",
   "task_location": "workspace_tasks",
-  "need_gpu": false,
-  "gpu": "preferred",
+  "need_gpu": true,
+  "gpu": "required",
   "icon": "https://i.imgur.com/HZCM0QM.png",
   "icon_cover": true,
   "context_menu": {