Closed
Description
When i run this code
a = torch.Tensor(5, 3) # construct a 5x3 matrix, uninitialized
b = torch.Tensor(5, 3) # construct a 5x3 matrix, uninitialized
if torch.cuda.is_available():
aa= a.cuda()
bb = b.cuda()
aa + bb
I get the following error message
RuntimeError: cuda runtime error (8) : invalid device function at /data/users/soumith/miniconda2/conda-bld/pytorch-0.1.6_1484802121799/work/torch/lib/THC/generated/../generic/THCTensorMathPointwise.cu:246
Metadata
Metadata
Assignees
Labels
No labels