albanD:

CUDA_LAUNCH_BLOCKING=1 python your_script.py.


And then, I upload my program to FloydHub.The error disappeared.What is the reason of it?My nvidia GPU is GTX 1050.I know it’s memory is small.However, The error don’t tell me it’s the memory problem. It’s weird.