I got this Runtime CUDA error how can i fix this?

Hi,
Illegal Memory Access is usually an internal error, which might have some dependencies on system configs, can you try shifting your environment on colab, and try if it works?