environments Training documentation - Azure/azureml-assets GitHub Wiki

Training

Environments in this category

  • acft-hf-nlp-data-import

    Environment used by Hugging Face NLP Finetune components

  • acft-hf-nlp-gpu

    Environment used by Hugging Face NLP Finetune components

  • acft-mmdetection-image-gpu

    Environment used by MMDetection Image Finetune components

  • acft-mmtracking-video-gpu

    Environment used by MMTracking Video Finetune components

  • acft-multimodal-gpu

    Environment used by Multimodal classification Finetune components

  • acft-transformers-image-gpu

    Environment used by HuggingFace Transformers Image Finetune components

  • acpt-automl-image-framework-selector-gpu

    Environment used by framework selector component for automl image workloads

  • 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.2-cuda12.1

    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...

  • ai-ml-automl

    An environment used by Azure ML AutoML for training models.

  • ai-ml-automl-dnn

    An environment used by Azure ML AutoML for training models.

  • ai-ml-automl-dnn-forecasting-gpu

    An environment used by Azure ML AutoML for training models.

  • ai-ml-automl-dnn-gpu

    An environment used by Azure ML AutoML for training models.

  • ai-ml-automl-dnn-text-gpu

    An environment used by Azure ML AutoML for training models.

  • ai-ml-automl-dnn-text-gpu-ptca

    An environment used by Azure ML AutoML for training models.

  • ai-ml-automl-dnn-vision-gpu

    An environment used by Azure ML AutoML for training models.

  • ai-ml-automl-gpu

    An environment used by Azure ML AutoML for training models.

  • lightgbm-3.3

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

  • 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.

  • sklearn-1.5

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

  • tensorflow-2.16-cuda11

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

  • tensorflow-2.16-cuda12

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

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