environments data labeling sam - Azure/azureml-assets GitHub Wiki

data-labeling-sam

Overview

GPU based environment for AML Data Labeling SAM Embedding Generation.

Version: 18

Tags

OS : Ubuntu20.04 Python : 3.9 GPU : Cuda11

View in Studio: https://ml.azure.com/registries/azureml/environments/data-labeling-sam/version/18

Docker image: mcr.microsoft.com/azureml/curated/data-labeling-sam:18

Docker build context

Dockerfile

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

WORKDIR /

ENV CONDA_PREFIX=/azureml-envs/data-labeling-sam
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

# support Deepspeed launcher requirement of passwordless ssh login
RUN apt-get update
RUN apt-get install -y openssh-server openssh-client
⚠️ **GitHub.com Fallback** ⚠️