You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hi :) i run the infer_deeplab.sh setting --no-cuda but i get this error:
RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.
Can you help me? Thank you
I added map_location=torch.device('cpu') in torch.load in loader.py
The text was updated successfully, but these errors were encountered:
Hi :) i run the infer_deeplab.sh setting --no-cuda but i get this error:
RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.
Can you help me? Thank you
I added map_location=torch.device('cpu') in torch.load in loader.py
The text was updated successfully, but these errors were encountered: