Java Deep Learning Cookbook

CPU/GPU-specific configuration for training

Hardware-specific changes are generic configurations that can't be ignored in a distributed environment. DL4J supports GPU-accelerated training in NVIDIA GPUs with CUDA/cuDNN enabled. We can also perform Spark distributed training using GPUs.

In this recipe, we will configure CPU/GPU-specific changes.