PyTorch

Docker Hardened Image

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

pytorch/pytorch

pytorch

PyTorch is a deep learning framework that puts Python first.

rocm/pytorch

AMD

The PyTorch on ROCm Docker images

bitnamicharts/pytorch

VMware

Bitnami Helm chart for PyTorch

openeuler/pytorch

openeuler

The repository for openEuler PyTorch container images.

pytorch/almalinux-builder

pytorch

General Purpose image with conda installed to be used in PyTorch CI/CD

pytorch/torchserve-nightly

pytorch

https://github.com/pytorch/serve