ReadMeFeed / Community knowledge

#tensor

Real questions. Useful conversations. Find the people who know your stack.

Question1 answer· Accepted1 comment0 followingPosted Jun 280 views0 votes

Pytorch fails with CUDA error: device-side assert triggered on Colab

I am trying to initialize a tensor on Google Colab with GPU enabled. device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') t = torch.tensor([1,2], device=device) But I am getting this strange error. RuntimeError: CUDA error: device-side assert triggered CUDA kernel errors might be asynchronously reported at some other API call,so the stacktra

AI & Data

google-colaboratory