diff --git a/dkube/sdk/internal/dkube_api/models/job_status_model.py b/dkube/sdk/internal/dkube_api/models/job_status_model.py
index 16970799..530fa0a7 100644
--- a/dkube/sdk/internal/dkube_api/models/job_status_model.py
+++ b/dkube/sdk/internal/dkube_api/models/job_status_model.py
@@ -70,7 +70,7 @@ def state(self, state):
         :param state: The state of this JobStatusModel.  # noqa: E501
         :type: str
         """
-        allowed_values = ["QUEUED", "STARTING", "TRAINING", "RUNNING", "COMPLETE", "STOPPED", "STOPPING", "DELETING", "ERROR", "IMAGEBUILDINPROGRESS", "UPDATING"]  # noqa: E501
+        allowed_values = ["QUEUED", "STARTING", "TRAINING", "RUNNING", "COMPLETE", "STOPPED", "STOPPING", "DELETING", "ERROR", "IMAGEBUILDINPROGRESS", "UPDATING", "Waiting for 1 gpu(s)", "waiting for training to begin"]  # noqa: E501
         if state not in allowed_values:
             raise ValueError(
                 "Invalid value for `state` ({0}), must be one of {1}"  # noqa: E501
diff --git a/dkube/sdk/rsrcs/ide.py b/dkube/sdk/rsrcs/ide.py
index 324dd8bc..4864e964 100644
--- a/dkube/sdk/rsrcs/ide.py
+++ b/dkube/sdk/rsrcs/ide.py
@@ -43,9 +43,9 @@
 
 class DkubeIDE(object):
 
-    FRAMEWORK_OPTS = ["custom", "tensorflow_1.14", "tensorflow_2.0.0", "tensorflow_2.3.0",
-                      "tensorflow_r-1.14", "tensorflow_r-2.0.0",
-                      "pytorch_1.6", "sklearn_0.23.2"]
+    FRAMEWORK_OPTS = ["custom", "tensorflow_1.14", "tensorflow_1.14-gpu", "tensorflow_2.0.0", "tensorflow_2.0.0-gpu", "tensorflow_2.3.0", "tensorflow_2.3.0-gpu",
+                      "tensorflow_r-1.14", "tensorflow_r-1.14-gpu", "tensorflow_r-2.0.0", "tensorflow_r-2.0.0-gpu",
+                      "pytorch_1.6", "pytorch_1.6-gpu", "sklearn_0.23.2"]
 
     def __init__(self, user, name=generate('notebook'), description='', tags=[]):
         self.repo = JobInputDatumModel  # class assignment, caller creates objects
diff --git a/dkube/sdk/rsrcs/training.py b/dkube/sdk/rsrcs/training.py
index f8b3b36e..a41da715 100644
--- a/dkube/sdk/rsrcs/training.py
+++ b/dkube/sdk/rsrcs/training.py
@@ -47,9 +47,9 @@
 
 class DkubeTraining(object):
 
-    FRAMEWORK_OPTS = ["custom", "tensorflow_1.14", "tensorflow_2.0.0", "tensorflow_2.3.0",
-                      "tensorflow_r-1.14", "tensorflow_r-2.0.0",
-                      "pytorch_1.6", "sklearn_0.23.2"]
+    FRAMEWORK_OPTS = ["custom", "tensorflow_1.14", "tensorflow_1.14-gpu", "tensorflow_2.0.0", "tensorflow_2.0.0-gpu", "tensorflow_2.3.0", "tensorflow_2.3.0-gpu",
+                      "tensorflow_r-1.14", "tensorflow_r-1.14-gpu", "tensorflow_r-2.0.0", "tensorflow_r-2.0.0-gpu",
+                      "pytorch_1.6", "pytorch_1.6-gpu", "sklearn_0.23.2"]
 
     DISTRIBUTION_OPTS = ["manual", "auto"]