GPU-enabled Job Agent

Before you create your own GPU-enabled Job Agent, be aware that Altair RapidMiner provides pre-configured Docker images for Deep Learning. Nevertheless, if you want more technical detail, continue reading!

The main points are the following:

  • Your Job Agent must be installed on a computer with CUDA-compatible GPU.
  • You must install the Nvidia library CUDA and should install the library cuDNN for enhanced performance.

Create a GPU-enabled Job Agent

A Job Agent can take advantage of a GPU for processing images or training and scoring neural networks. Currently, one GPU can be used per Job Agent.

Take the following steps:

  1. Install the Job Agent on a computer that has a CUDA-compatible GPU.

  2. Follow the installation instructions for CUDA 10.1 (and cuDNN version 7.6).