DL4J实战之四：经典卷积实例(GPU版本)，java进阶路线图

• 使用CPU还是GPU，具体操作步骤非常简单：切换不同的依赖库即可，下面分别介绍

• 如果您用CPU做训练，则依赖库和版本如下：

org.deeplearning4j

deeplearning4j-core

1.0.0-beta6

org.nd4j

nd4j-native

1.0.0-beta6

如果您用GPU做训练，且CUDA版本是9.2，则依赖库和版本如下：

org.deeplearning4j

deeplearning4j-core

1.0.0-beta6

org.deeplearning4j

deeplearning4j-cuda-9.2

1.0.0-beta6

org.nd4j

nd4j-cuda-9.2-platform

1.0.0-beta6

• java代码就不在这里贴出了，用的是[《DL4J实战之三：经典卷积实例(LeNet-5)》](

)中的代码，不做任何改变

[](

)内存设置

• 使用IDEA运行代码的时候，可以按照当前硬件情况将内存适当调大，步骤如下图：