Enabling Cloudera Data Science Workbench to use GPUs

To enable GPU usage on Cloudera Data Science Workbench, perform the following steps to provision the Cloudera Data Science Workbench hosts. As noted in the following instructions, certain steps must be repeated on all gateway hosts that have GPU hardware installed on them.

The steps described in this document have been tested and validated on the following setup:
Custom Engine CDSW NVIDIA Driver

cuda-engine:10

- Based on CDSW engine:13

- Ships CUDA 10.1

1.8.x, 1.9.x

418.39 (or higher)

NVIDIA/CUDA compatibility matrix

For more compatibility information across NVIDIA Drivers and CUDA, refer the NVIDIA documentation: CUDA Compatibility.