I am trying to run my DNN model based on PyTorch framework which requires CUDA support for image classification/real-time inference since I am getting this error message in RTMaps:
RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on device: cpu
Is there any way I will be able to run my model for image classification on NXP Bluebox 2.0 (S32V234) with CUDA support?
Thank you for your time and advise.