bubbliiiing / mask-rcnn-tf2

这是一个mask-rcnn-tf2的库,可以用于训练自己的模型。
MIT License
64 stars 9 forks source link

博主你好,train.py在epoch2训练结束后程序运行结束,显示 an illegal memory access was encountered #32

Open Chester818 opened 1 year ago

Chester818 commented 1 year ago

博主你好,我的运行环境是windows,显卡:RTX A6000 , CUDA 11.8,cuDNN 8.7.0, tensorflow-gpu 2.6.0,batch size为1。在训练到epoch2结束后程序运行结束,以下是报错的代码: 2023-03-14 16:05:09.195263: E tensorflow/stream_executor/cuda/cuda_event.cc:29] Error polling for event status: failed to query event: CUDA_ERROR_ILLEGAL_ADDRESS: an illegal memory access was encountered 2023-03-14 16:05:09.196247: F tensorflow/core/common_runtime/device/device_event_mgr.cc:221] Unexpected Event status: 1 请问该如何解决,谢谢博主!