DL4J中文文档/配置/GPU/CPU设置

用于GPU和CPU的ND4J后端

你可以通过更改ND4J的POM.xml文件中的依赖项来为后端线性代数操作选择GPU或本地CPU。你的选择将影响应用程序中正在使用的ND4J和DL4J。

如果你的CUDA v9.2+已经安装并且有NVIDIA兼容的硬件，那么你的依赖声明将看起来像：

 org.nd4j
 nd4j-cuda-9.2
 1.0.0-beta2

image.gif

否则，你将需要使用ND4J的本地实现作为CPU后端：

 org.nd4j
 nd4j-native
 1.0.0-beta2

[图片上传中...(image-3ae026-1540642042555-1)]

系统架构

如果你在多个操作系统/系统架构上开发项目，则可以在artifactId的末尾添加-platform，该artifactId将下载大多数主要系统的二进制文件。

 ...
 nd4j-native-platform
 ...

image.gif

多GPU

如果你有多个GPU但你的系统迫使你 只能用一个，你可以用 helper CudaEnvironment.getInstance().getConfiguration().allowMultiGPU(true);作为你的 main()方法的第一行。

CuDNN

查看我们的 CuDNN 页。

CUDA 安装

在NVIDIA 网站 上查看CUDA安装说明

翻译：风一样的男子

image

如果您觉得我的文章给了您帮助，请为我买一杯饮料吧！以下是我的支付宝，意思一下我将非常感激！

image