Azure Machine Learning Curated Environments

This article lists the curated environments in Azure Machine Learning. Curated environments are provided by Azure Machine Learning and are available in your workspace by default. They are backed by cached Docker images that use the latest version of the Azure Machine Learning SDK, reducing the run preparation cost and allowing for faster deployment time. Use these environments to quickly get started with various machine learning frameworks.

Note

This list is updated as of April 2021. Use the Python SDK or CLI to get the most updated list of environments and their dependencies. For more information, see the environments article. Following the release of this new set, previous curated environments will be hidden but can still be used.

PyTorch

  • AzureML-pytorch-1.7-ubuntu18.04-py37-cuda11-gpu
    • An environment for deep learning with PyTorch containing the Azure ML SDK and additional python packages.
    • PyTorch version: 1.7
    • Python version: 3.7
    • Base image: mcr.microsoft.com/azureml/openmpi4.1.0-cuda11.0.3-cudnn8-ubuntu18.04
    • CUDA version: 11.0.3
    • OpenMPI version: 4.1.0
    • Ubuntu version: 18.04

Sklearn

  • AzureML-sklearn-0.24-ubuntu18.04-py37-cuda11-gpu
    • An environment for tasks such as regression, clustering, and classification with Scikit-learn. Contains the Azure ML SDK and additional python packages.
    • Scikit-learn version: 24.1
    • Python version: 3.7
    • Base image: mcr.microsoft.com/azureml/openmpi4.1.0-cuda11.0.3-cudnn8-ubuntu18.04
    • CUDA version: 11.0.3
    • OpenMPI version: 4.1.0
    • Ubuntu version: 18.04

TensorFlow

  • AzureML-tensorflow-2.4-ubuntu18.04-py37-cuda11-gpu
    • An environment for deep learning with Tensorflow containing the Azure ML SDK and additional python packages.
    • Tensorflow version: 2.4
    • Python version: 3.7
    • Base image: mcr.microsoft.com/azureml/openmpi4.1.0-cuda11.0.3-cudnn8-ubuntu18.04
    • CUDA version: 11.0.3
    • OpenMPI version: 4.1.0
    • Ubuntu version: 18.04

Inference only curated environments and prebuilt docker images