RuntimeError: Attempted to set the storage of a tensor on device "cpu" to a storage on different device "cuda:0". This is no longer allowed; the devices must match.

[liujiaoshou17]

RuntimeError: Attempted to set the storage of a tensor on device "cpu" to a storage on different device "cuda:0". This is no longer allowed; the devices must match.
使用预训练模型为imagenet时会报错，其他预训练模型就可以

[bayuezha]

在 hub.py里修改最后一行，删去, map_location=map_location

return torch.load(cached_file, weights_only=weights_only)