Backends
Hardware setup for Eclipse Deeplearning4j, including GPUs and CUDA.
Last updated
Was this helpful?
Hardware setup for Eclipse Deeplearning4j, including GPUs and CUDA.
Last updated
Was this helpful?
You can choose GPUs or native CPUs for your backend linear algebra operations by changing the dependencies in ND4J's POM.xml file. Your selection will affect both ND4J and DL4J being used in your application.
If you have CUDA v9.2+ installed and NVIDIA-compatible hardware, then your dependency declaration will look like:
As of now, the artifactId
for the CUDA versions can be one of nd4j-cuda-9.2
, nd4j-cuda-10.0
, nd4j-cuda-10.1
or nd4j-cuda-10.2
.
You can also find the available CUDA versions via or in the .
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 -platform
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 .