CUDA error: device-side assert triggered after a certain steps

ankitb131313
Wed, 31 Jan 2024 03:04:49 +0000

I was able to get past this error by setting PYTORCH_USE_CUDA_DSA as “1”

os.environ["PYTORCH_USE_CUDA_DSA"] = "1"

torch version is 2.2.0