RuntimeError: Attempting to deserialize object on CUDA device 3 but torch.cuda.device_count() is 1. Please use torch.load with map_location to map your storages to an existing device.
https://github.com/computationalmedia/semstyle/issues/3
RuntimeError: Attempting to deserialize object on CUDA device 2 but torch.cuda.device_count() is 1. Please use torch.load with m
After running python img_to_text.py --test_folder photos, I'm getting File "/home/ahrzb/.pyenv/versions/2.7.15/envs/mzh2.7/lib/python2.7/site-packages/torch/serialization.py", line 35...
github.com
loaded_state = torch.load(model_path+seq_to_seq_test_model_fname,map_location='cuda:0')
여러 대의 gpu로 training 된 경우, 저렇게 mapping location을 따로 언급해줘야함!
'에러모음집' 카테고리의 다른 글
| E: Unable to acquire the dpkg frontend lock (/var/lib/dpkg/lock-frontend), is another process using it? (0) | 2021.09.06 |
|---|---|
| pytorchstreamreader 에러 (0) | 2021.09.06 |
| HOROVOD ver.2 (0) | 2021.09.04 |
| good code vs bad code (0) | 2021.09.04 |
| pytorch memory profiler (0) | 2021.09.03 |