Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False

 

Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False

 

此时改为:

torch.load("0.9472_0048.weights",map_location='cpu')

你可能感兴趣的:(pytorch知识宝典)