PyTorch
dhi.io/pytorch
A python-based machine learning framework, providing tensors, dynamic neural networks and strong GPU acceleration.
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:
For more information, visit https://pytorch.org/docs/.
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.
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.
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
Distribution:
Packages
239
Vulnerabilities
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.
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
Join GitHub Discussions or our Slack community to share ideas, ask questions, and connect with the team.
Go to discussionsJoin community