Closed zhuhu00 closed 1 year ago
Hi,
Can you check what is the value of device when you run it? Usually it should be cuda or cpu, as:
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
If it works as expected, could you then check what would happen if you force all the tensors as .cuda() or .cpu()? It looks like the device of your machine is not found correctly.
Thanks, I forgot to modify it when i try to test on multi-gpu before. It's solved now.
hi, jianyuan This error occurred when I ran your code. Is this normal?