Enabling Trainers with GPUs in On-Premise Podman Deployments

  • Updated on Dec 2, 2025
  • Published on Sep 17, 2024
  • 4 minute(s) read
Prev Next

With Hyperscience, you can classify and extract data from Unstructured documents. However, automating these processes requires more computing resources than our automation capabilities for Structured and Semi-structured documents do.

In order to process Unstructured documents in on-premise deployments of Hyperscience, you need to add a trainer that has both a GPU (graphics processing unit) and a CPU (central processing unit). GPUs have specialized cores that allow the system to perform multiple computations in parallel, reducing the time required to complete the complex operations required to train models for Unstructured documents. When you attach a trainer whose machine has a GPU, you can maximize the benefits of Long-form Extraction. To learn more about this feature, see Long-form Extraction.

Support for GPUs in Hyperscience

  • Trainers with GPUs are supported in Docker, Podman, and Kubernetes deployments of Hyperscience v41 and later. Use of GPUs with v40 or earlier is not supported.

  • A trainer with a GPU is required to train Long-form Extraction models.

  • While GPUs are not required to train Field Identification or Table Identification models, you can use a trainer with a GPU to train these models.

  • The training of other types of models with a GPU is not supported. However, you can train them with a CPU on a trainer machine that has both a GPU and a CPU.

  • Machines with GPUs are not supported for the Hyperscience application.

This article describes how to enable a trainer with both a GPU and a CPU in an on-premise Docker deployment of Hyperscience. Steps 1-3 must be completed before untarring the Hyperscience bundle on the trainer machine. For more information on setting up the trainer, see Technical Installation / Upgrade Instructions.

1. Make sure your GPU hardware meets the requirements.

See Infrastructure Requirements for more information.

2. Make sure your trainer meets the software compatibility requirements.

There are several software-compatibility considerations to keep in mind when setting up your trainer.

a. Verify that the lspci command is enabled.

To do so:

  1. Install the pciutils package by running the following command on RHEL:

    yum -y install pciutils

  2. Run lspci to make sure the command has been enabled.

b. Verify that your GPU supports CUDA.

CUDA is a parallel computing platform and programming model created by NVIDIA. Machine learning often uses CUDA-based libraries, SDKs, and other tools.

You can find out whether your GPU supports CUDA by running the following command:

lspci | grep -i nvidia

For more information, see NVIDIA’s CUDA GPUs - Compute Capability and NVIDIA CUDA Installation Guide for Linux.

c. Verify that you have a supported version of Linux.

Follow the instructions in NVIDIA’s NVIDIA CUDA Installation Guide for Linux to check your version of Linux. Then, make sure your Linux version is supported by the latest CUDA Toolkit by reviewing NVIDIA’s NVIDIA CUDA Toolkit Release Notes.  

d. Verify that the system has gcc installed.

The gcc compiler is required for development using the CUDA Toolkit. To make sure it is installed, follow the instructions in NVIDIA’s NVIDIA CUDA Installation Guide for Linux.

e. Verify that the system has the current Kernel headers and development packages installed.

Kernel headers are header files that specify the interface between the Linux kernel and userspace libraries and programs. The CUDA driver requires that the kernel headers and development packages for the running version of the kernel be installed at the time of the driver installation, as well whenever the driver is rebuilt. For example, if your system is running kernel version 3.17.4-301, the 3.17.4-301 kernel headers and development packages must also be installed.

To verify that these requirements are met, run the following command:

sudo dnf install kernel-devel-$(uname -r) kernel-headers-$(uname -r)

For more information and commands for various Linux distributions, see NVIDIA’s NVIDIA CUDA Installation Guide for Linux.

3. Install the CUDA Driver and NVIDIA Container Toolkit.

  1. Ensure your system meets the prerequisites for the driver installation:

  2. Install the driver by running the following command on RHEL:

    sudo dnf module install nvidia-driver:latest-dkms

  3. Install the container toolkit by completing the steps in the "Installing with Yum or Dnf" section of NVIDIA's Installing the NVIDIA Container Toolkit.

  4. Configure Podman to use NVIDIA devices in the container:

    1. Complete the steps in the "Procedure" section of NVIDIA's Support for Container Device Interface.

    2. Edit /usr/share/containers/containers.conf:

      1. Set NVIDIA as the runtime (i.e., runtime=”nvidia"). If there is any other runtime set, comment it out.

      2. Add nvidia=[“/usr/bin/nvidia-container-runtime”] to [engine.runtimes].
        After completing these steps, your file should look similar to the one shown below. 

        runtime = "nvidia"

...

[engine.runtimes]

...

nvidia = [

"/usr/bin/nvidia-container-runtime",

]

  5. (Optional) Reboot the system:

    sudo reboot

While we recommend installing the latest version of the CUDA driver, only the minimum required version for your version of Hyperscience is required. See Infrastructure Requirements for more information.

Next steps

After you’ve enabled your trainer, follow the steps in “Long-form Extraction” ( v41 | v42 ) to apply the trainer to your Long-form Extraction use case.