search

ardigen / MAT

The official implementation of the Molecule Attention Transformer.
MIT License
239 stars 57 forks source link

CUDA error: an illegal memory access was encountered #14

Open SK124 opened 4 years ago

SK124 commented 4 years ago

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