Model mgmt torch net libxml2 (#4204)

* Update Dockerfile

* Update requirements.txt

* Update Dockerfile

* Update Dockerfile

Author: raghu-microsoft

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

@@ -2,7 +2,7 @@ FROM mcr.microsoft.com/aifx/acpt/stable-ubuntu2204-cu118-py310-torch222:{{latest
 
 WORKDIR /
 
-RUN apt-get update && apt-get upgrade -y && apt-get install wget nscd python3-idna git git-lfs -y
+RUN apt-get update && apt-get upgrade -y && apt-get install wget nscd python3-idna git git-lfs libxml2 -y
 
 # azcopy install starts
 RUN echo "Downloading azcopy to file azcopy.tar ....";\
@@ -25,7 +25,7 @@ RUN conda install python=3.9 -y
 RUN pip install -r requirements.txt --no-cache-dir
 
 # Vulnerability fix
-RUN pip install Pillow gunicorn onnx==1.17.0 idna tqdm requests==2.32.1 tornado==6.4.2 certifi==2024.07.04 urllib3==1.26.19 scikit-learn==1.5.1 mlflow==2.20.3 mlflow-skinny==2.20.3 marshmallow==3.23.2
+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

@@ -15,3 +15,17 @@ pyarrow==14.0.2
 GitPython=={{latest-pypi-version}}
 paramiko=={{latest-pypi-version}}
 cryptography==44.0.2
+####
+Pillow 
+gunicorn 
+onnx==1.17.0 
+idna 
+tqdm 
+requests==2.32.1 
+tornado==6.4.2 
+certifi==2024.07.04 
+urllib3==1.26.19 
+scikit-learn==1.5.1 
+mlflow==2.20.3 
+mlflow-skinny==2.20.3 
+marshmallow==3.23.2