NGC | Catalog
Logo for PyG
PyG (PyTorch Geometric) is a library built upon PyTorch to easily write and train Graph Neural Networks (GNNs) for a wide range of applications related to structured data.
Latest Tag
February 1, 2024
Compressed Size
10.56 GB
Multinode Support
Multi-Arch Support
24.01-py3 (Latest) Security Scan Results

Linux / arm64

Sorry, your browser does not support inline SVG.

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 an introductory example about training a GCN, try python /workspace/examples/

See /workspace/ for details

The container also uses torch-geometric 2.4.0 & pyg-lib 0.2.0. This container also contains the GNN Platform (/opt/pyg/gnn-platform), an NVIDIA project that provides a low-code API for rapid GNN experimentation and training/deploying end-to-end GNN pipelines. Examples can be found at /workspace/gnn-platform-examples. For more details about the GNN Platform, see /opt/pyg/gnn-platform/

This container is built on NVIDIA PyTorch container (see contents of PyTorch container)

Key Features in this release:

  • torch-frame integration

  • NVIDIAs syngen tool for synthetic graph data generation. See for details

  • torch.compile accelerations.

    • We recommend using torch.compile on your GNN models for accelerating any example effortlessly.
    • Example: model = torch.compile(model)


There are two main prerequisites for this container:

Running the container

Use the following commands to run the container, where <xx.xx> is the container version.

docker run --gpus all -it --rm

For example, 23.11 for November 2023 release:

docker run --gpus all -it --rm