时间戳

首页 > 解决方案 > 如何在 DeepLearning4J 中的 GPU 上创建 ND4J NDArrays？

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

问题描述

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

例如在 pytorch 中，

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

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

标签： javapytorchdeeplearning4jdl4j

解决方案

您所要做的就是在您的 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

推荐阅读