/DL4J_practice

DL4J practice

Primary LanguageJavaApache License 2.0Apache-2.0

DL4J_practice

DL4J practice, simple MNIST classificaiton.

I modified the pom.xml dependency to fit CUDA 11.2 & cudnn cuDNN v8.1.1

and all run well under my Ampere architecture GPU 3060.

Driver detail

image