environments acpt automl image framework selector gpu

acpt-automl-image-framework-selector-gpu

Overview

Environment used by framework selector component for automl image workloads

Version: 57

Tags

Preview

View in Studio: https://ml.azure.com/registries/azureml/environments/acpt-automl-image-framework-selector-gpu/version/57

Docker image: mcr.microsoft.com/azureml/curated/acpt-automl-image-framework-selector-gpu:57

Docker build context

Dockerfile

# PTCA image
FROM mcr.microsoft.com/aifx/acpt/stable-ubuntu2204-cu118-py310-torch260:biweekly.202506.1.v1

RUN apt-get -y update && apt-get install -y expat

# Install required packages
COPY requirements.txt .
RUN pip install -r requirements.txt --no-cache-dir

# Flag needed to enable control flow which is in PrP.
ENV AZURE_ML_CLI_PRIVATE_FEATURES_ENABLED=True

# Vulnerability Fix
RUN pip install gunicorn==23.0.0
RUN pip install certifi==2024.07.04
RUN pip install onnx==1.17.0