DL4J中文文檔/配置/CuDNN

與cuDNN一起使用DL4J DL4J支持CUDA,但可以進一步通過cuDNN加速。大多數2D 卷積神經網絡層(如ConvolutionLayer、SubsamplingLayer等),以及LSTM和BatchNormalization層都支持cuDNN。 爲了可以使用cuDNN,你首先要切換ND4J到CUDA後端。這可以通過在pom.xml文件中用nd4j-cuda-8.0, nd4j-cuda
相關文章
相關標籤/搜索