environments acft group relative policy optimization - Azure/azureml-assets GitHub Wiki
Environment used by Group Relative Policy Optimization component
Version: 5
Preview
View in Studio: https://ml.azure.com/registries/azureml/environments/acft-group-relative-policy-optimization/version/5
Docker image: mcr.microsoft.com/azureml/curated/acft-group-relative-policy-optimization:5
#PTCA image
FROM mcr.microsoft.com/aifx/acpt/stable-ubuntu2204-cu126-py310-torch271:biweekly.202507.2
USER root
RUN pip install --upgrade pip
COPY requirements.txt .
RUN pip install -r requirements.txt --no-cache-dir
RUN pip install azureml-acft-common-components==0.0.77
RUN pip install numpy==2.2.5
RUN pip install azureml-evaluate-mlflow==0.0.77
RUN pip install mlflow==3.1.0
RUN pip install transformers==4.52.1
# Upgrade requests in the system Python (3.13) for fixing vulnerability
RUN /opt/conda/bin/python3.13 -m pip install --upgrade requests urllib3 || true
# clean conda and pip caches
RUN rm -rf ~/.cache/pip