I am getting this:
nore667e@taurusi8023:~/bigdl_pipeline> spark-submit --conf spark.driver.memory=400g --driver-java-options='-Dorg.bytedeco.javacpp.logger.debug=true' --class com.examples.DeepLearningOnSpark.mnist_image.streaming_approach.MnistRetrainingMain target/deepLearningSimpleOne-1.0-SNAPSHOT-jar-with-dependencies.jar
22/09/02 19:03:00 WARN NativeCodeLoader: Unable to load native-hadoop library for your platform... using builtin-java classes where applicable
Debug: Loading class org.bytedeco.javacpp.presets.javacpp
Debug: Loading class org.bytedeco.javacpp.Loader
Debug: Loading /home/h4/nore667e/.javacpp/cache/deepLearningSimpleOne-1.0-SNAPSHOT-jar-with-dependencies.jar/org/bytedeco/javacpp/linux-x86_64/libjnijavacpp.so
Debug: Loading class org.bytedeco.javacpp.presets.javacpp
Debug: Loading class org.bytedeco.javacpp.Pointer
Warning: Versions of org.bytedeco:javacpp:1.5.7 and org.bytedeco:cuda:11.4-8.2-1.5.6 do not match.
Debug: Loading class org.bytedeco.javacpp.presets.javacpp
Debug: Loading class org.bytedeco.cuda.global.cudart
Debug: Loading library cudart
Debug: Loading /home/h4/nore667e/.javacpp/cache/deepLearningSimpleOne-1.0-SNAPSHOT-jar-with-dependencies.jar/org/bytedeco/cuda/linux-x86_64/libjnicudart.so
Debug: Loading class org.bytedeco.javacpp.presets.javacpp
Debug: Loading class org.bytedeco.cuda.global.cudart
Debug: Loading class org.bytedeco.cuda.global.cublas
Debug: Loading class org.bytedeco.javacpp.presets.javacpp
Debug: Loading class org.bytedeco.cuda.global.cudart
Debug: Loading class org.bytedeco.cuda.global.cublas
Debug: Loading library cublasLt
Debug: Loading library cublas
Debug: Loading /home/h4/nore667e/.javacpp/cache/deepLearningSimpleOne-1.0-SNAPSHOT-jar-with-dependencies.jar/org/bytedeco/cuda/linux-x86_64/libjnicublas.so
22/09/02 19:03:02 INFO Nd4jBackend: Loaded [JCublasBackend] backend
Debug: Loading class org.nd4j.linalg.jcublas.bindings.Nd4jCuda
Debug: Loading library curand
Debug: Loading library cusolver
Debug: Loading library cusparse
Debug: Loading library cudnn
Debug: Failed to load for cudnn@.8: java.lang.UnsatisfiedLinkError: no cudnn in java.library.path
Debug: Loading library cudnn_ops_infer
Debug: Failed to load for cudnn_ops_infer@.8: java.lang.UnsatisfiedLinkError: no cudnn_ops_infer in java.library.path
Debug: Loading library cudnn_ops_train
Debug: Failed to load for cudnn_ops_train@.8: java.lang.UnsatisfiedLinkError: no cudnn_ops_train in java.library.path
Debug: Loading library cudnn_adv_infer
Debug: Failed to load for cudnn_adv_infer@.8: java.lang.UnsatisfiedLinkError: no cudnn_adv_infer in java.library.path
Debug: Loading library cudnn_adv_train
Debug: Failed to load for cudnn_adv_train@.8: java.lang.UnsatisfiedLinkError: no cudnn_adv_train in java.library.path
Debug: Loading library cudnn_cnn_infer
Debug: Failed to load for cudnn_cnn_infer@.8: java.lang.UnsatisfiedLinkError: no cudnn_cnn_infer in java.library.path
Debug: Loading library cudnn_cnn_train
Debug: Failed to load for cudnn_cnn_train@.8: java.lang.UnsatisfiedLinkError: no cudnn_cnn_train in java.library.path
Debug: Loading /home/h4/nore667e/.javacpp/cache/deepLearningSimpleOne-1.0-SNAPSHOT-jar-with-dependencies.jar/org/nd4j/linalg/jcublas/bindings/linux-x86_64/libgomp.so.1
Debug: Loading /home/h4/nore667e/.javacpp/cache/deepLearningSimpleOne-1.0-SNAPSHOT-jar-with-dependencies.jar/org/nd4j/linalg/jcublas/bindings/linux-x86_64/libnd4jcuda.so
Debug: Loading /home/h4/nore667e/.javacpp/cache/deepLearningSimpleOne-1.0-SNAPSHOT-jar-with-dependencies.jar/org/nd4j/linalg/jcublas/bindings/linux-x86_64/libjnind4jcuda.so
22/09/02 19:03:03 INFO NativeOpsHolder: Number of threads used for linear algebra: 32
Exception in thread "main" java.lang.ExceptionInInitializerError
at com.examples.DeepLearningOnSpark.mnist_image.streaming_approach.MnistRetrainingMain.entryPoint(MnistRetrainingMain.java:27)
at com.examples.DeepLearningOnSpark.mnist_image.streaming_approach.MnistRetrainingMain.main(MnistRetrainingMain.java:22)
at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)
at sun.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:62)
at sun.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:43)
at java.lang.reflect.Method.invoke(Method.java:498)
at org.apache.spark.deploy.JavaMainApplication.start(SparkApplication.scala:52)
at org.apache.spark.deploy.SparkSubmit.org$apache$spark$deploy$SparkSubmit$$runMain(SparkSubmit.scala:928)
at org.apache.spark.deploy.SparkSubmit.doRunMain$1(SparkSubmit.scala:180)
at org.apache.spark.deploy.SparkSubmit.submit(SparkSubmit.scala:203)
at org.apache.spark.deploy.SparkSubmit.doSubmit(SparkSubmit.scala:90)
at org.apache.spark.deploy.SparkSubmit$$anon$2.doSubmit(SparkSubmit.scala:1007)
at org.apache.spark.deploy.SparkSubmit$.main(SparkSubmit.scala:1016)
at org.apache.spark.deploy.SparkSubmit.main(SparkSubmit.scala)
Caused by: java.lang.RuntimeException: java.lang.NullPointerException
at org.nd4j.linalg.factory.Nd4j.initWithBackend(Nd4j.java:5205)
at org.nd4j.linalg.factory.Nd4j.initContext(Nd4j.java:5070)
at org.nd4j.linalg.factory.Nd4j.<clinit>(Nd4j.java:290)
... 14 more
Caused by: java.lang.NullPointerException
at org.nd4j.linalg.factory.Nd4j.initWithBackend(Nd4j.java:5148)
... 16 more
22/09/02 19:03:03 INFO ShutdownHookManager: Shutdown hook called
22/09/02 19:03:03 INFO ShutdownHookManager: Deleting directory /tmp/spark-133284bb-7c9f-442d-8d5e-c5ea2dfd650e