environments PyTorch documentation

PyTorch

Environments in this category

---

• acft-group-relative-policy-optimization

  Environment used by Group Relative Policy Optimization component

• acft-hf-nlp-gpu

  Environment used by Hugging Face NLP Finetune components

• acft-medimageinsight-adapter-finetune

  AzureML ACFT MedImageInsight Adapter Image for Training

• acft-medimageinsight-embedding

  Environment used by MedImageInsight Embedding Finetune Component

• acft-medimageinsight-embedding-generator

  Environment used by MedImageInsight Embedding Generation component

• acft-medimageparse-finetune

  AzureML ACFT MedImageParse Image for Training

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

• acpt-pytorch-2.8-cuda12.6

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

• designer-pytorch-2.3-train

  AzureML Designer PyTorch Train image