tboo/ludwig_gpu repository overview

A docker container that allows to easily run Ludwig with GPU support. Could also be used on multi-GPU machines for distributed training.

Based on the NVIDIA NGC docker image nvcr.io/nvidia/tensorflow:19.07-py3 and modified by adding/updating

Ludwig 0.2

OpenMPI 4.0.1

Driver Requirements

Release 19.07 is based on NVIDIA CUDA 10.1.168, which requires NVIDIA Driver release 418.67. However, if you are running on Tesla (Tesla V100, Tesla P4, Tesla P40, or Tesla P100), you may use NVIDIA driver release 384.111+ or 410. The CUDA driver's compatibility package only supports particular drivers. For a complete list of supported drivers, see the CUDA Application Compatibility topic. F

GPU Requirements

Release 19.07 supports CUDA compute capability 6.0 and higher. This corresponds to GPUs in the Pascal, Volta, and Turing families.

Tested on Ubuntu 18.04 LTS with NVIDIA driver v.430