automl-dnn-vision-gpu update (#3269)

Azure · Aug 19, 2024 · cf49f14 · cf49f14
1 parent 1f537bd
commit cf49f14
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/assets/training/vision/environments/automl-dnn-vision-gpu/context/Dockerfile b/assets/training/vision/environments/automl-dnn-vision-gpu/context/Dockerfile
@@ -1,4 +1,5 @@
-FROM mcr.microsoft.com/azureml/openmpi4.1.0-cuda11.6-cudnn8-ubuntu20.04:{{latest-image-tag}}
+FROM mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu121-py310-torch222:{{latest-image-tag}}
+
 
 ENV AZUREML_CONDA_ENVIRONMENT_PATH /azureml-envs/azureml-automl-dnn-vision-gpu
 # Prepend path to AzureML conda environment
@@ -10,6 +11,17 @@ ENV MLFLOW_MODEL_FOLDER="mlflow-model"
 # ENV AML_APP_ROOT="/var/mlflow_resources"
 # ENV AZUREML_ENTRY_SCRIPT="mlflow_score_script.py"
 
+# Inference requirements
+COPY --from=mcr.microsoft.com/azureml/o16n-base/python-assets:20230419.v1 /artifacts /var/
+RUN /var/requirements/install_system_requirements.sh && \
+    cp /var/configuration/rsyslog.conf /etc/rsyslog.conf && \
+    cp /var/configuration/nginx.conf /etc/nginx/sites-available/app && \
+    ln -sf /etc/nginx/sites-available/app /etc/nginx/sites-enabled/app && \
+    rm -f /etc/nginx/sites-enabled/default
+ENV SVDIR=/var/runit
+ENV WORKER_TIMEOUT=400
+EXPOSE 5001 8883 8888
+
 ENV ENABLE_METADATA=true
 
 # Create conda environment

diff --git a/assets/training/vision/environments/automl-dnn-vision-gpu/context/conda_dependencies.yaml b/assets/training/vision/environments/automl-dnn-vision-gpu/context/conda_dependencies.yaml
@@ -27,4 +27,5 @@ dependencies:
   - azureml-interpret=={{latest-pypi-version}}
   - azureml-train-automl-runtime=={{latest-pypi-version}}
   - azureml-automl-dnn-vision=={{latest-pypi-version}}
-  - azureml-dataprep>=2.24.4
+  - azureml-dataprep>=2.24.4
+  - azure-identity>=1.16.1