- 再看看CUDA 9.1对应的nd4j库的版本情况,如下图红框,最新的是2018年的1.0.0-beta,与核心库差距太大了:
- 好了,再来看看CUDA 9.2对应的nd4j库的版本情况,如下图红框,最新的是1.0.0-beta6,与核心库差两个版本,因此,建议使用CUDA 9.2:
[](()使用GPU的具体操作步骤
-
使用CPU还是GPU,具体操作步骤非常简单:切换不同的依赖库即可,下面分别介绍
-
如果您用CPU做训练,则依赖库和版本如下:
org.deeplearning4j
deeplearning4j-core
1.0.0-beta6
org.nd4j
nd4j-native
1.0.0-beta6
如果您用GPU做训练,且CUDA版本是9.2,则依赖库