Updating the torch image in Dockerfile (#4220)

* Update Dockerfile

* 4021541

* 4021541

* 4021541

* 4021541

---------

Co-authored-by: utkshukla <utkshukla@microsoft.com>

Author: raghu-microsoft

assets/training/model_management/environments/model-management/context/Dockerfile

@@ -1,4 +1,4 @@
-FROM mcr.microsoft.com/aifx/acpt/stable-ubuntu2204-cu118-py310-torch222:{{latest-image-tag:biweekly\.\d{6}\.\d{1}.*}}
+FROM mcr.microsoft.com/aifx/acpt/stable-ubuntu2204-cu118-py310-torch260:{{latest-image-tag:biweekly\.\d{6}\.\d{1}.*}}
 
 WORKDIR /
 
@@ -16,17 +16,12 @@ RUN echo "Downloading azcopy to file azcopy.tar ....";\
 
 COPY requirements.txt .
 
-# Switch to python==3.9 to prevent numpy==1.23.5 build failure
-# numpy==1.23.5 is a hard dependency in azureml-automl-dnn-vision
-# which is not compatible with py310.
-# Going forward  azureml-automl-dnn-vision needs to update version for torch and numpy
-RUN conda install python=3.9 -y
+# This line is added because azureml-automl-dnn-vision depends on python 3.9.
+# Added ignore python requires to get this installed.
+RUN pip install --ignore-requires-python azureml-automl-dnn-vision
 
 RUN pip install -r requirements.txt --no-cache-dir
 
-# Vulnerability fix
-RUN pip install torch==2.6.0 torchvision --no-cache-dir
-
 # List pip packages
 RUN pip list
 

assets/training/model_management/environments/model-management/context/requirements.txt

@@ -10,7 +10,7 @@ wget=={{latest-pypi-version}}
 applicationinsights=={{latest-pypi-version}}
 azureml-automl-core=={{latest-pypi-version}}
 azureml-telemetry=={{latest-pypi-version}}
-azureml-automl-dnn-vision=={{latest-pypi-version}}
+#azureml-automl-dnn-vision=={{latest-pypi-version}}
 pyarrow==14.0.2
 GitPython=={{latest-pypi-version}}
 paramiko=={{latest-pypi-version}}
@@ -22,7 +22,7 @@ onnx==1.17.0
 idna 
 tqdm 
 requests==2.32.1 
-tornado==6.4.2 
+tornado==6.5
 certifi==2024.07.04 
 urllib3==1.26.19 
 scikit-learn==1.5.1