[Linux server error] TypeError: can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor

The reason for this error is that numpy is used to calculate the tensor in the previous cuda. Check the code and use A.data.cpu().numpy() to transform it to continue

Guess you like

Origin blog.csdn.net/qq_45347185/article/details/108965347