Jetson Nano 2GB运行YOLOv4-tiny目标跟踪代码时TensorFlow出现CPU到GPU张量复制错误的排查求助

我在Jetson Nano 2GB设备上尝试运行一个目标跟踪仓库代码：Windows环境下按照TensorFlow官方文档配置依赖后，代码可正常运行；但在Jetson Nano上完成依赖配置后，执行以下命令时出现报错：

python3 object_tracker.py --weights ./checkpoints/yolov4-tiny-416 --model yolov4 --video 0 --tiny --info

错误信息如下：

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]

我怀疑是内存问题，通过tegrastats监控设备状态，发现RAM最高使用1338MB（总1980MB），SWAP仅使用121MB（总11230MB），内存仍有剩余，但错误依旧。

目前已尝试以下方案，均未解决问题：

• 更换其他目标检测仓库
• 将TensorFlow版本切换为代码要求的2.3.0
• 训练模型后通过TensorRT运行

恳请各位提供技术帮助。

内容的提问来源于stack exchange，提问作者Prithviraj Kanaujia