diff --git a/assets/responsibleai/environments/responsibleai-text/context/Dockerfile b/assets/responsibleai/environments/responsibleai-text/context/Dockerfile
index 64fe8a2b7f..ffc184fccc 100644
--- a/assets/responsibleai/environments/responsibleai-text/context/Dockerfile
+++ b/assets/responsibleai/environments/responsibleai-text/context/Dockerfile
@@ -42,6 +42,9 @@ RUN pip install git+https://github.com/huggingface/evaluate.git
 RUN pip install 'Werkzeug>=3.0.3'
 RUN pip install 'tqdm>=4.66.3'
 
+# To resolve CVE-2024-5480 vulnerability issue for torch < 2.2.2
+RUN pip install 'torch>=2.2.2'
+
 RUN pip freeze
 
 # This is needed for mpi to locate libpython
diff --git a/assets/responsibleai/environments/responsibleai-text/context/conda_dependencies.yaml b/assets/responsibleai/environments/responsibleai-text/context/conda_dependencies.yaml
index be898ff726..ee7922a994 100644
--- a/assets/responsibleai/environments/responsibleai-text/context/conda_dependencies.yaml
+++ b/assets/responsibleai/environments/responsibleai-text/context/conda_dependencies.yaml
@@ -26,4 +26,3 @@ dependencies:
     - datasets
     - typing-extensions>=4.8.0
     - gevent-websocket
-    - torch>=2.2.2
diff --git a/assets/responsibleai/environments/responsibleai-text/tests/requirements.txt b/assets/responsibleai/environments/responsibleai-text/tests/requirements.txt
index 676eb8b8c9..43002541d0 100644
--- a/assets/responsibleai/environments/responsibleai-text/tests/requirements.txt
+++ b/assets/responsibleai/environments/responsibleai-text/tests/requirements.txt
@@ -1,3 +1,3 @@
-azure-ai-ml==0.1.0b4
+azure-ai-ml~=1.21.0
 azure.identity==1.10.0
-azureml-core==1.43.0
\ No newline at end of file
+azureml-core>=1.56.0
diff --git a/assets/responsibleai/environments/responsibleai-text/tests/responsibleai_text_sample_test.py b/assets/responsibleai/environments/responsibleai-text/tests/responsibleai_text_sample_test.py
index 38b5a43e78..0750307f76 100644
--- a/assets/responsibleai/environments/responsibleai-text/tests/responsibleai_text_sample_test.py
+++ b/assets/responsibleai/environments/responsibleai-text/tests/responsibleai_text_sample_test.py
@@ -7,7 +7,7 @@
 from pathlib import Path
 from azure.ai.ml import MLClient
 from azure.ai.ml import command
-from azure.ai.ml._restclient.models import JobStatus
+from azure.ai.ml.operations._run_history_constants import JobStatus
 from azure.ai.ml.entities import Environment, BuildContext
 from azure.identity import AzureCliCredential
 
diff --git a/assets/responsibleai/environments/responsibleai-vision/context/conda_dependencies.yaml b/assets/responsibleai/environments/responsibleai-vision/context/conda_dependencies.yaml
index 4543751f78..e323b0bd3a 100644
--- a/assets/responsibleai/environments/responsibleai-vision/context/conda_dependencies.yaml
+++ b/assets/responsibleai/environments/responsibleai-vision/context/conda_dependencies.yaml
@@ -26,5 +26,4 @@ dependencies:
     - ipython<8.13
     - typing-extensions>=4.8.0
     - gevent-websocket
-    - torch>=2.2.2
     - opencv-python-headless==4.8.1.78
diff --git a/assets/responsibleai/environments/responsibleai-vision/tests/requirements.txt b/assets/responsibleai/environments/responsibleai-vision/tests/requirements.txt
index 676eb8b8c9..a46f576cea 100644
--- a/assets/responsibleai/environments/responsibleai-vision/tests/requirements.txt
+++ b/assets/responsibleai/environments/responsibleai-vision/tests/requirements.txt
@@ -1,3 +1,3 @@
-azure-ai-ml==0.1.0b4
+azure-ai-ml~=1.13.0
 azure.identity==1.10.0
-azureml-core==1.43.0
\ No newline at end of file
+azureml-core>=1.56.0