com.facebook.jni.CppException: Could not run 'aten::empty_strided' with arguments from the 'CUDA' backend.

FATAL EXCEPTION: main
Process: org.pytorch.demo.imagesegmentation, PID: 5452
java.lang.RuntimeException: Unable to start activity ComponentInfo{org.pytorch.demo.imagesegmentation/org.pytorch.imagesegmentation.MainActivity}: com.facebook.jni.CppException: Could not run ‘aten::empty_strided’ with arguments from the ‘CUDA’ backend. This could be because the operator doesn’t exist for this backend, or was omitted during the selective/custom build process (if using custom build). If you are a Facebook employee using PyTorch on mobile, please visit Internal Login for possible resolutions. ‘aten::empty_strided’ is only available for these backends: [Dense, Fake, Negative, UNKNOWN_TENSOR_TYPE_ID, UNKNOWN_TENSOR_TYPE_ID, SparseCPU, SparseCUDA, SparseHIP, UNKNOWN_TENSOR_TYPE_ID, UNKNOWN_TENSOR_TYPE_ID, UNKNOWN_TENSOR_TYPE_ID, SparseVE, UNKNOWN_TENSOR_TYPE_ID, UNKNOWN_TENSOR_TYPE_ID].

                                                         CPU: registered at /data/users/atalman/pytorch/build_android_arm64-v8a/aten/src/ATen/RegisterCPU.cpp:37386 [kernel]
                                                         Vulkan: registered at /data/users/atalman/pytorch/aten/src/ATen/native/vulkan/ops/Factory.cpp:47 [kernel]
                                                         QuantizedCPU: registered at /data/users/atalman/pytorch/build_android_arm64-v8a/aten/src/ATen/RegisterQuantizedCPU.cpp:1294 [kernel]
                                                         BackendSelect: registered at /data/users/atalman/pytorch/build_android_arm64-v8a/aten/src/ATen/RegisterBackendSelect.cpp:726 [kernel]
                                                         ADInplaceOrView: fallthrough registered at /data/users/atalman/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:64 [backend fallback]
                                                         AutogradOther: fallthrough registered at /data/users/atalman/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:35 [backend fallback]
                                                         AutogradCPU: fallthrough registered at /data/users/atalman/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:39 [backend fallback]
                                                         AutogradCUDA: fallthrough registered at /data/users/atalman/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:47 [backend fallback]
                                                         AutogradXLA: fallthrough registered at /data/users/atalman/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:51 [backend fallback]
                                                         AutogradMPS: fallthrough registered at /data/users/atalman/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:59 [backend fallback]
                                                         AutogradXPU: fallthrough registered at /data/users/atalman/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:43 [backend fallback]
                                                         AutogradHPU: fallthrough registered at /data/users/atalman/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:68 [backend fallback]
                                                         AutogradLazy: fallthrough registered at /data/users/atalman/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:55 [backend fallback]
                                                         Functionalize: registered at /data/users/atalman/pytorch/aten/src/ATen/FunctionalizeFallbackKernel.cpp:89 [backend fallback]
                                                          ()
                                                         Exception raised from reportError at /data/users/atalman/pytorch/aten/src/ATen/core/dispatch/OperatorEntry.cpp:447 (most recent call first):
                                                         (no backtrace available)
                                                         	at android.app.ActivityThread.performLaunchActivity(ActivityThread.java:3762)
                                                         	at android.app.ActivityThread.handleLaunchActivity(ActivityThread.java:3929)
                                                         	at android.app.servertransaction.LaunchActivityItem.execute(LaunchActivityItem.java:103)
                                                         	at android.app.servertransaction.TransactionExecutor.executeCallbacks(TransactionExecutor.java:135)
                                                         	at android.app.servertransaction.TransactionExecutor.execute(TransactionExecutor.java:95)
                                                         	at android.app.ActivityThread$H.handleMessage(ActivityThread.java:2284)
                                                         	at android.os.Handler.dispatchMessage(Handler.java:106)
                                                         	at android.os.Looper.loopOnce(Looper.java:210)
                                                         	at android.os.Looper.loop(Looper.java:299)
                                                         	at android.app.ActivityThread.main(ActivityThread.java:8302)
                                                         	at java.lang.reflect.Method.invoke(Native Method)
                                                         	at com.android.internal.os.RuntimeInit$MethodAndArgsCaller.run(RuntimeInit.java:556)
                                                         	at com.android.internal.os.ZygoteInit.main(ZygoteInit.java:1037)