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:
Install the Job Agent on a computer that has a CUDA-compatible GPU.
Follow the installation instructions for CUDA 10.1 (and cuDNN version 7.6).