PyTorch Production Branch 6 offers a 9-month lifecycle for API stability, with monthly patches for high and critical software vulnerabilities. This release includes Government Ready images for regulated environments.
What Is PyTorch?
PyTorch is a GPU-accelerated tensor computational framework that offers a high degree of flexibility and speed for deep learning. It integrates seamlessly with popular Python libraries such as NumPy, SciPy, and Cython, extending its functionality to meet the diverse needs of users. PyTorch also employs a tape-based system for automatic differentiation at both the functional and neural network layer level, ensuring accelerated NumPy-like functionality.
What Is PyTorch Production Branch 6?
The PyTorch Production Branch, exclusively available with NVIDIA AI Enterprise, is a 9-month supported, API-stable branch that includes monthly fixes for high and critical software vulnerabilities. This branch provides a stable and secure environment for building your mission-critical AI applications. The PyTorch production branch releases every six months with a three-month overlap in between two releases.
Getting started with PyTorch Production Branch
Before you start, ensure that your environment is set up by following one of the deployment guides available in the NVIDIA AI Enterprise Documentation.
For an overview of the features included in the PyTorch Production Branch 6, please refer to the Release Notes for PyTorch 26.03.
For a comprehensive collection of resources on PyTorch, including tutorials, documentation, and examples, visit the following links:
Additionally, if you're looking for information on Docker containers and guidance on running a container, review the Containers For Deep Learning Frameworks User Guide.
Government Ready: STIG/FIPS Hardening
This ensures the highest level of security and compliance for regulated environments, the x86 container image for this branch is:
-
STIG Ubuntu 24.04 hardened
-
Supports FIPS 140-2 / 3 validated crypto / uses libraries that support FIPS crypto
Learn more about NVIDIA's hardened image in the AI Software for Regulated Environments White Paper.
Compatible Infrastructure Software Versions
For the optimized performance, it is highly recommended to deploy the supported NVIDIA AI Enterprise Infrastructure software in conjunction with your AI software. Production Branch 6 is compatible with NVIDIA AI Enterprise Infrastructure 7.
Security Vulnerabilities in Open Source Packages
Please review the Security Scanning tab to view the latest security scan results.
For certain open-source vulnerabilities listed in the scan results, NVIDIA provides a response in the form of a Vulnerability Exploitability eXchange (VEX) document. The VEX information can be reviewed and downloaded from the Security Scanning tab.
Get Help
Enterprise Support
Get access to knowledge base articles and support cases or submit a ticket.
NVIDIA AI Enterprise Documentation
Visit the NVIDIA AI Enterprise Documentation Hub for release documentation, deployment guides and more.
NVIDIA Licensing Portal
Go to the NVIDIA Licensing Portal to manage your software licenses.
Governing Terms
The software and materials are governed by the NVIDIA Software and Model Evaluation License.