Pytorch with apex training

pytorch with nvidia-apex

PyTorch

Docker Hardened Image

A python-based machine learning framework, providing tensors, dynamic neural networks and strong GPU acceleration.

acuvity/apex-agent

Acuvity Inc.

APEX agent runs the proxy and routes all AI traffic through the Acuvity APEX proxy.

acuvity/cloud-apex

Acuvity Inc.

Standalone cloud proxy for AI traffic interception and policy enforcement without privileged access.

rocm/pytorch

AMD

The PyTorch on ROCm Docker images

bitnamicharts/pytorch

VMware

Bitnami Helm chart for PyTorch

rocm/pytorch-nightly

AMD

This docker repo is used for hosting pytorch-nightly-build docker images.

rocm/pytorch-xdit

AMD

Public repo for ROCm-based xDiT images

rocm/pytorch-training

AMD

Unified PyTorch base container optimized for training with ROCm

intel/intel-optimized-pytorch

Intel Corporation

Containers for running PyTorch workloads on Intel® Architecture.