CUDA error: unknown error

[Mauricio]

Hi,

I am having this problem with pytorch... any solution?

import torch
a = torch.rand(5, 3)
device = torch.device('cuda')
a.to(device)
Traceback (most recent call last):
 File "<stdin>", line 1, in <module>
RuntimeError: CUDA error: unknown error

Thank you..
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://mailman.srv.cs.cmu.edu/pipermail/autonlab-users/attachments/20180905/58e205f3/attachment.html>