Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

how to use this repo in cpu machine #56

Open
Ironman1508 opened this issue Aug 10, 2020 · 0 comments
Open

how to use this repo in cpu machine #56

Ironman1508 opened this issue Aug 10, 2020 · 0 comments

Comments

@Ironman1508
Copy link

can someone guide how to use this repo in a cpu machine. I'm facing this error while i'm running on CPU Machine

(lidar) C:\Users\Aravind Giri\Complex-YOLOv3>python test_detection.py --split=sample --folder=sampledata
Namespace(class_path='data/classes.names', conf_thres=0.5, folder='sampledata', img_size=608, model_def='config/complex_tiny_yolov3.cfg', nms_thres=0.5, split='sample', weights_path='checkpoints/tiny-yolov3_ckpt_epoch-220.pth')
Traceback (most recent call last):
File "test_detection.py", line 106, in
model.load_state_dict(torch.load(opt.weights_path))
File "C:\Users\Aravind Giri.conda\envs\lidar\lib\site-packages\torch\serialization.py", line 585, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "C:\Users\Aravind Giri.conda\envs\lidar\lib\site-packages\torch\serialization.py", line 765, in _legacy_load
result = unpickler.load()
File "C:\Users\Aravind Giri.conda\envs\lidar\lib\site-packages\torch\serialization.py", line 721, in persistent_load
deserialized_objects[root_key] = restore_location(obj, location)
File "C:\Users\Aravind Giri.conda\envs\lidar\lib\site-packages\torch\serialization.py", line 174, in default_restore_location
result = fn(storage, location)
File "C:\Users\Aravind Giri.conda\envs\lidar\lib\site-packages\torch\serialization.py", line 150, in _cuda_deserialize
device = validate_cuda_device(location)
File "C:\Users\Aravind Giri.conda\envs\lidar\lib\site-packages\torch\serialization.py", line 134, in validate_cuda_device
raise RuntimeError('Attempting to deserialize object on a CUDA '
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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant