environments acft transformers image gpu

acft-transformers-image-gpu

Overview

Environment used by HuggingFace Transformers Image Finetune components

Version: 33

Tags

Preview

View in Studio: https://ml.azure.com/registries/azureml/environments/acft-transformers-image-gpu/version/33

Docker image: mcr.microsoft.com/azureml/curated/acft-transformers-image-gpu:33

Docker build context

Dockerfile

# PTCA image
FROM mcr.microsoft.com/aifx/acpt/stable-ubuntu2004-cu118-py310-torch222:biweekly.202405.3

USER root
RUN apt-get -y update

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

# Vulnerability Fix
RUN pip install gunicorn==22.0.0

# dummy number to change when needing to force rebuild without changing the definition: 1