Model can't be moved to GPU
#20
by ftopal - opened
When I move model to GPU, I get RuntimeError
RuntimeError: CUDA error: device-side assert triggered
CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be
incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.
What is the issue here?
I could not reproduce your reported issue. Are you running other code before this code snippet?
Nope, I found that it happens with other models as well so I suppose something Google Colab specific going here. I can't figure it out what tho. We can probably close this one as it doesn't seem model specific in this case.
ftopal changed discussion status to closed