environments tensorflow 2.11 cuda11 - Azure/azureml-assets GitHub Wiki

tensorflow-2.11-cuda11

Overview

An environment for deep learning with Tensorflow containing the Azure ML SDK and additional python packages.

Version: 30

Tags

Tensorflow : 2.11 GPU : Cuda11 OS : Ubuntu20.04 Training Preview Python : 3.8

View in Studio: https://ml.azure.com/registries/azureml/environments/tensorflow-2.11-cuda11/version/30

Docker image: mcr.microsoft.com/azureml/curated/tensorflow-2.11-cuda11:30

Docker build context

Dockerfile

FROM mcr.microsoft.com/azureml/openmpi4.1.0-cuda11.3-cudnn8-ubuntu20.04:20240712.v1

WORKDIR /

ENV CONDA_PREFIX=/azureml-envs/tensorflow-2.11-cuda11
ENV CONDA_DEFAULT_ENV=$CONDA_PREFIX
ENV PATH=$CONDA_PREFIX/bin:$PATH

# Create conda environment
COPY conda_dependencies.yaml .
RUN HOROVOD_WITH_TENSORFLOW=1 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

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

⚠️ **GitHub.com Fallback** ⚠️