cuda error

I get the error when I use scripts/edges/batch_hed.py to detect image edges

F0912 15:12:50.764644 23187 syncedmem.cpp:19] Check failed: error == cudaSuccess (29 vs. 0)  driver shutting down

my work enviroment is :ubuntu16 ,cuda8.0
does anybody know how to solve this probrom and get this script to work correct?