Hi,
I try running this simple block of code:
list_torch = [1,2,3,4]
tenso = torch.tensor(list_torch).view(-1,1)
tenso = tenso.to(device=‘cuda’)
I get this error:
RuntimeError                              Traceback (most recent call last)
 in ()
----> 1 tenso = tenso.to(device=‘cuda’)
RuntimeError: CUDA error: device-side assert triggered
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
The code works fine in CPU though, it throws error in COLAB as well as in my local machine