RuntimeError: CUDA error: device-side assert triggered

Question

I\'m newbie to pytorch. So I practice how to use Pytorch. I ran some code in Pytorch book, but I met errors

I\'m using python==3.7

pytorch==1.7

cuda==10.1