RuntimeError: CUDA error: device-side assert triggered

[Karthikeyan-Ram]

while running the following code in colab pro (either GPU or TPU), getting the error.
"RuntimeError: CUDA error: device-side assert triggered"

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
data = data.to(device)

[rusty1s]

Do you have a small script to reproduce?