DL4J中文文档/配置/CuDNN

与cuDNN一起使用DL4J DL4J支持CUDA,但可以进一步通过cuDNN加速。大多数2D 卷积神经网络层(如ConvolutionLayer、SubsamplingLayer等),以及LSTM和BatchNormalization层都支持cuDNN。 为了可以使用cuDNN,你首先要切换ND4J到CUDA后端。这可以通过在pom.xml文件中用nd4j-cuda-8.0, nd4j-cuda
相关文章
相关标签/搜索