It seems that the problem is related to the type of GPU ？
I test the same code on two desktop computer.
On the one with Intel UHD 630 and NVidia GT 730 GPU, the code reports error.
while on the one with Intel UHD 750 and NVidia RTX A4000 GPU, the code works fine.