environments PyTorch documentation

Pytorch

Environments in this category

---

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

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

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

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

• acpt-pytorch-1.11-cuda11.5

  DEPRECATED: Please use acpt-pytorch-1.11-cuda11.3 instead.

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

• acft-hf-nlp-gpu

  Environment used by Hugging Face NLP Finetune components