environments Training documentation

Training

Environments in this category

---

• acpt-pytorch-1.11-cuda11.3

  Recommended environment for Deep Learning with PyTorch on Azure containing the Azure ML SDK with the latest compatible versions of Ubuntu, Python, PyTorch, CUDA\RocM, combined with optimizers like ORT Training,+DeepSpeed+MSCCL+ORT MoE and more. The image introduces preview of new fastcheckpointin...

• acpt-pytorch-1.12-cuda11.6

  Recommended environment for Deep Learning with PyTorch on Azure containing the Azure ML SDK with the latest compatible versions of Ubuntu, Python, PyTorch, CUDA\ROCm, combined with optimizers like ORT Training,+DeepSpeed+MSCCL+ORT MoE and more. The image introduces preview of new fastcheckpointin...

• acpt-pytorch-1.12-py39-cuda11.6

  Recommended environment for Deep Learning with PyTorch on Azure containing the Azure ML SDK with the latest compatible versions of Ubuntu, Python, PyTorch, CUDA\RocM, combined with optimizers like ORT Training,+DeepSpeed+MSCCL+ORT MoE and more. The image introduces preview of new fastcheckpointin...

• acpt-pytorch-1.13-cuda11.7

  Recommended environment for Deep Learning with PyTorch on Azure containing the Azure ML SDK with the latest compatible versions of Ubuntu, Python, PyTorch, CUDA\RocM, combined with optimizers like ORT Training,+DeepSpeed+MSCCL+ORT MoE and more. The image introduces preview of new fastcheckpointin...

• acpt-pytorch-2.0-cuda11.7

  Recommended environment for Deep Learning in public preview with PyTorch on Azure containing the Azure ML SDK with the latest compatible versions of Ubuntu, Python, PyTorch, CUDA\RocM, combined with optimizers like ORT Training,+DeepSpeed+MSCCL+ORT MoE and more. The image introduces newly release...

• lightgbm-3.3

  An environment for tasks such as regression, clustering, and classification with LightGBM. Contains the Azure ML SDK and additional python packages.

• pytorch-1.11-ubuntu20.04-py38-cuda11-gpu

  DEPRECATED: Please use pytorch-1.11-cuda11.5 instead.

• sklearn-1.0

  An environment for tasks such as regression, clustering, and classification with Scikit-learn. Contains the Azure ML SDK and additional python packages.

• sklearn-1.1

  An environment for tasks such as regression, clustering, and classification with Scikit-learn. Contains the Azure ML SDK and additional python packages.

• tensorflow-2.11-cuda11

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

• tensorflow-2.12-cuda11

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

• tensorflow-2.8-cuda11

  DEPRECATED: Please use tensorflow-2.11-cuda11 or tensorflow-2.12-cuda11 instead. An environment for deep learning with Tensorflow containing the Azure ML SDK and additional python packages.

• tensorflow-2.9-cuda11

  DEPRECATED: Please use tensorflow-2.11-cuda11 or tensorflow-2.12-cuda11 instead. An environment for deep learning with Tensorflow containing the Azure ML SDK and additional python packages.