If you have CUDA v9.2+ installed and NVIDIA-compatible hardware, then your dependency declaration will look like:
As of now, the for the CUDA versions can be one of nd4j-cuda-9.0
, or nd4j-cuda-10.0
.
Otherwise you will need to use the native implementation of ND4J as a CPU backend:
If you are developing your project on multiple operating systems/system architectures, you can add to the end of your artifactId
which will download binaries for most major systems.
See our page on .
Check the NVIDIA guides for instructions on setting up CUDA on the NVIDIA website.