environments designer pytorch 2.3

designer-pytorch-2.3

Overview

AzureML Designer PyTorch image

Version: 1

Tags

Python : 3.8.10 OS : Ubuntu20.04

View in Studio: https://ml.azure.com/registries/azureml/environments/designer-pytorch-2.3/version/1

Docker image: mcr.microsoft.com/azureml/curated/designer-pytorch-2.3:1

Docker build context

Dockerfile

FROM mcr.microsoft.com/azureml/openmpi4.1.0-ubuntu20.04:20241020.v1

WORKDIR /

ENV CONDA_PREFIX=/azureml-envs/designer-pytorch-2.3
ENV CONDA_DEFAULT_ENV=$CONDA_PREFIX
ENV PATH=$CONDA_PREFIX/bin:$PATH

# This is needed for mpi to locate libpython
ENV LD_LIBRARY_PATH=$CONDA_PREFIX/lib:$LD_LIBRARY_PATH


# Create conda environment
COPY conda_dependencies.yaml .
RUN conda env create -p $CONDA_PREFIX -f conda_dependencies.yaml -q && \
    rm conda_dependencies.yaml && \
    conda run -p $CONDA_PREFIX pip cache purge && \
    conda clean -a -y