You can replace the <artifactId> ... </artifactId>, depending on your preference:

nd4j-cuda-$CUDA_VERSION (where CUDA_VERSION is one of 7.5 or 8.0)

That’s it.

If you have several GPUs, but your system is forcing you to use just one, there’s a solution. Just add CudaEnvironment.getInstance().getConfiguration().allowMultiGPU(true); as first line of your main() method.

Check the NVIDIA guides for instructions on setting up CUDA on Linux, Windows, and OSX.