PyTorch

dhi.io/pytorch

PyTorch

CIS
linux/arm64
linux/amd64

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

About PyTorch

PyTorch is an open-source machine learning framework that provides a flexible and powerful platform for building deep learning models. It offers dynamic computational graphs, automatic differentiation, and native GPU acceleration through CUDA. PyTorch is widely used in research and production for tasks ranging from computer vision to natural language processing and reinforcement learning.

Key features include:

  • Dynamic computational graphs for flexible model building
  • Strong GPU acceleration with CUDA support
  • Extensive ecosystem including TorchVision for computer vision and TorchAudio for audio processing
  • TorchScript for production deployment
  • Native support for distributed training

For more information, visit https://pytorch.org/docs/.

About Docker Hardened Images

Docker Hardened Images are built to meet the highest security and compliance standards. They provide a trusted foundation for containerized workloads by incorporating security best practices from the start.

Why use Docker Hardened Images?

These images are published with zero-known CVEs, include signed provenance, and come with a complete Software Bill of Materials (SBOM) and VEX metadata. They're designed to secure your software supply chain while fitting seamlessly into existing Docker workflows.

Trademarks

PyTorch, the PyTorch logo and any related marks are trademarks of The Linux Foundation.

NVIDIA®, the NVIDIA® logo, and CUDA® are trademarks and/or registered trademarks of NVIDIA Corporation in the U.S. and other countries. Other company and product names may be trademarks of the respective companies with which they are associated.

This listing is prepared by Docker. All third-party product names, logos, and trademarks are the property of their respective owners and are used solely for identification. Docker claims no interest in those marks, and no affiliation, sponsorship, or endorsement is implied.

Try DHI Enterprise

Start a free 30-day DHI Enterprise trial to mirror this image to your organization's registry and unlock full benefits.

SLA-backed CVE remediations

FIPS & STIG-compliant variants

Customizations at enterprise scale

Try DHI Enterprise
Security summary

Distribution:

debian 13

Packages

239

Vulnerabilities

0
0
9
4
13
View full security details
7 Tools included

python

The Python interpreter.

pip

Python's package installer.

torch

PyTorch core library for tensor computation and deep learning.

torchvision

Computer vision library with datasets, models, and image transformations for PyTorch.

torchaudio

Audio processing library with datasets, transforms, and models for PyTorch.

torchelastic

Elastic training library for distributed PyTorch training with fault tolerance.

triton

OpenAI Triton GPU programming language and JIT compiler for high-performance deep learning kernels.

More info

Recently pushed images

PyTorch 2.11.x (dev)

Updated: 3 hours ago

PyTorch 2.11.x

Updated: 3 hours ago

PyTorch 2.11.x (dev)

Updated: 4 hours ago

PyTorch 2.11.x

Updated: 3 hours ago

PyTorch 2.11.x (dev)

Updated: 4 hours ago

View all
Got questions or feedback about this image?

Join GitHub Discussions or our Slack community to share ideas, ask questions, and connect with the team.

Go to discussionsJoin community