I am facing memory errors while converting my model to tensor RT I have tensorflow 2.4 installed on my jetson Nano with cuda 10, RAM of 4GB and swap memory of 8gb available.
I am trying to convert custom yolo V3 to protect single class object detection model to tensor RT. Jetson Nano is running out of memory again and again the same the same script running on running on my PC with GPU and on Google colab also which was able to successfully convert yolo V3 to tensor RT. but as Nvidia documentation says that these model are hardware specific hands and cannot deploy those model on jetson Nano.
Here is the error
tensorflow.python.framework.errors_impl.InternalError: Failed copying input tensor from /job:localhost/replica:0/task:0/device:CPU:0 to /job:localhost/replica:0/task:0/device:GPU:0 in order to run Identity: Dst tensor is not initialized. [Op:Identity]
2021-03-15 15:02:30.619804: I tensorflow/core/common_runtime/bfc_allocator.cc:1051] Stats: