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) |
For more compatibility information across NVIDIA Drivers and CUDA, refer the NVIDIA documentation: CUDA Compatibility.