Open leecheukwai opened 2 years ago
my cpu is i7 6700k ,did the error means these code cant run on my cpu?
If you read the code you will find that these operations only have GPU implementation. So just move your tensor to GPU: input = torch.randn(shape).cuda()
my cpu is i7 6700k ,did the error means these code cant run on my cpu?