pytorchTutorial icon indicating copy to clipboard operation
pytorchTutorial copied to clipboard

Published 20 hours ago • verified publisher icon patrickloeber

02_tensor_basics.py

Open gaushi opened this issue 3 years ago • 0 comments

Hi @python-engineer , on line 131, shouldn't we assign z again on CPU? z = z.to("cpu") before then calling z.numpy() I got following error without doing so:

TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

Thanks for all the great material! Cheers, Gautam

gaushi avatar Mar 31 '22 19:03 gaushi