掘金 后端 ( ) • 2021-10-15 11:37
再看看CUDA 9.1对应的nd4j库的版本情况,如下图红框,最新的是2018年的1.0.0-beta,与核心库差距太大了:。
好了,再来看看CUDA 9.2对应的nd4j库的版本情况,如下图红框,最新的是1.0.0-beta6,与核心库差两个版本,因此,建议使用CUDA 9.2:。
如果您用GPU做训练,且CUDA版本是9.2,则依赖库和版本如下: ... 阅读全文