如何在 DeepLearning4J 中的 GPU 上创建 ND4J NDArrays？

【问题标题】：How to create ND4J NDArrays on GPU in DeepLearning4J?如何在 DeepLearning4J 中的 GPU 上创建 ND4J NDArrays？
【发布时间】：2020-06-30 19:31:36
【问题描述】：

有没有办法在 DL4J 中创建 NDArray，以便在 GPU 上完成操作？

例如在pytorch中，

cuda0 = torch.device('cuda:0')
x = torch.tensor([1., 2.], device=cuda0)

如果我的系统配备 GPU，我找不到指定后端（cpu 或 gpu）的方法？

【问题讨论】：

标签： java pytorch deeplearning4j dl4j

【解决方案1】：

您所要做的就是在您的 pom.xml 中包含正确的 nd4j 后端版本。在这种情况下，通常是：

<dependency>
 <groupId>org.nd4j</groupId>
 <artifactId>nd4j-cuda-10.2</artifactId>
 <version>1.0.0-beta7</version>
</dependency>

这是撰写本文时最新的 dl4j 版本。确保在 maven central 上仔细检查。这将包括 cuda 10.2

在此处查看更多信息： https://deeplearning4j.konduit.ai/config/backends

【讨论】：

我需要将INDArray.data().pointer() 传递给我的 cpp 后端。而且我还需要使用NDArray构造函数在gpu上使用libnd4j创建NDArray