When running PyTorch scripts, we may encounter the following error: “CUDA error: device-side assert triggered”, causing a runtime error. In this tutorial, we will show you how to fix this error.
How To Fix This Error?
The best way is to run PyTorch scripts on the CPU rather than the GPU. It will report the location of the error in the script.
This error usually occurs when the shape of a PyTorch tensor is incorrect.
As an example:
- one_hot = torch.zeros_like(cosine)
- one_hot.scatter_(1, label.view(-1, 1), 1.0)
If you run this script on GPU, you will see the following:
- RuntimeError: CUDA error: device-side assert triggered
however, if you run the same script on the CPU, you will see this:
Now you check where this error occurred in your script and fix it accordingly.
Apart from this, it is good programming practice to check your code beforehand to ensure there are no errors. Try the fix given above and fix your error easily. We have a whole bunch of other content coming up too. Meanwhile, you can check out some of our other articles too.