What can I help you with?
NVIDIA Optimized Frameworks

PyG Release 25.03

This PyG container release is intended for use on the NVIDIA® Hopper Architecture GPU, NVIDIA H100, the NVIDIA® Ampere Architecture GPU, NVIDIA A100, and the associated NVIDIA CUDA® 12 and NVIDIA cuDNN 9 libraries.

Driver Requirements

Release 25.03 is based on CUDA 12.8.1.010 which requires NVIDIA Driver release 570 or later. However, if you are running on a data center GPU (for example, B100, L40, or any other data center GPU), you can use NVIDIA driver release 470.57 (or later R470), 525.85 (or later R525), 535.86 (or later R535), or 550.54 (or later R550) in forward-compatibility mode.

The CUDA driver's compatibility package only supports particular drivers. Thus, users should upgrade from all R418, R440, R450, R460, R510, R520, R530, R545 and R555 and R560 drivers, which are not forward-compatible with CUDA 12.8. For a complete list of supported drivers, see the CUDA Application Compatibility topic. For more information, see CUDA Compatibility and Upgrades.

Contents of the PyG container

This container image includes the complete source of the NVIDIA version of PyG in /opt/pyg/pytorch_geometric. It is prebuilt and installed as a system Python module. The /workspace/examples folder is copied from /opt/pyg/pytorch_geometric/examples for users starting to run PyG. For example, to run the gcn.py example:

Copy
Copied!
            

/workspace/examples# python gcn.py

See /workspace/README.md for details.

The container also includes the following:

GPU Requirements

Release 25.03 supports CUDA compute capability 6.0 and later. This corresponds to GPUs in the NVIDIA Pascal, NVIDIA Volta™, NVIDIA Turing™, NVIDIA Ampere architecture, and NVIDIA Hopper™ architecture families. For a list of GPUs to which this compute capability corresponds, see CUDA GPUs. For additional support details, see Deep Learning Frameworks Support Matrix.

Key Features and Enhancements

Announcements

  • None.

NVIDIA PyG Container Versions

The PyG container supports the same version of Ubuntu and CUDA as the PyTorch container.

Container Version Ubuntu CUDA Toolkit PyG PyTorch
25.03 24.04 NVIDIA CUDA 12.8.1.012 2.7.0 2.7.0a0+7c8ec84dab
25.01 NVIDIA CUDA 12.8.0 2.7.0 2.6.0a0+ecf3bae40a
24.11 NVIDIA CUDA 12.6.3 2.7.0 2.6.0a0+df5bbc0
24.09 22.04 NVIDIA CUDA 12.6.1 2.6.0 2.5.0a0+b465a5843b
24.07 NVIDIA CUDA 12.5.1 2.6.0 2.4.0a0+3bcc3cddb5
24.05 NVIDIA CUDA 12.4.1 2.6.0 2.4.0a0+07cecf4
24.03 NVIDIA CUDA 12.4.0.41 2.5.0 2.3.0a0+40ec155e58
24.01 NVIDIA CUDA 12.3.2 2.4.0 2.2.0a0+81ea7a4
23.11 NVIDIA CUDA 12.3.0 2.4.0 23.11
23.01 20.04 NVIDIA CUDA 12.0.1 2.2.0 23.01

Known Issues

  • None.

© Copyright 2025, NVIDIA. Last updated on May 8, 2025.