Java Deep Learning Cookbook

How it works...

We need to enable a proper ND4J backend so that we can utilize GPU resources, as we mentioned in step 1. Enable the nd4j-cuda-x.x dependency in your pom.xml file for GPU training, where x.x refers to the CUDA version that you have installed.

We may include both ND4J backends (CUDA/native dependencies) if the master node is running on the CPU and the worker nodes are running on the GPU, as we mentioned in the previous recipe. If both backends are present in the classpath, the CUDA backend will be tried out first. If it doesn't load for some reason, then the CPU backend (native) will be loaded. The priority can also be changed by changing the BACKEND_PRIORITY_CPU and BACKEND_PRIORITY_GPU environment variables in the master ...