CUDA error: an illegal memory access was encountered

[SK124]

When I run the cell model.cuda( ), I get the following error.
I am running it on Google Colab with GPU support ( Tesla K80 )

RuntimeError Traceback (most recent call last)
in ()
----> 1 model.cuda()

8 frames
/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py in (t)
456 Module: self
457 """
--> 458 return self._apply(lambda t: t.cuda(device))
459
460 def cpu(self: T) -> T:

RuntimeError: CUDA error: an illegal memory access was encountered