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

RunTimeError #45

Closed
ks322181 opened this issue Dec 8, 2019 · 6 comments
Closed

RunTimeError #45

ks322181 opened this issue Dec 8, 2019 · 6 comments

Comments

@ks322181
Copy link

ks322181 commented Dec 8, 2019

Runtimerror: Attempting to deserialize objects on 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 storage to cpu

I think there is a problem with torch and torchvision version

Can anyone help me?
Thanks in Advance.

@yexing99
Copy link

I got the same error but fixed it by obj_detect_state_dict = torch.load(_config['tracktor']['obj_detect_model'],map_location=torch.device('cpu'))

@timmeinhardt
Copy link
Collaborator

This is indeed an issue of running the code on CPU and our models being trained on GPU. I just pushed a change which should resolve this issue and is similar to @yexing99 solution.

@timmeinhardt
Copy link
Collaborator

Can we close this issue?

@ks322181
Copy link
Author

I am working on GPU system. Where i have cuda version 8. But still i am facing the same issue. Is it compatible with cuda 8. If not can you specific which cuda version it will support.

@timmeinhardt
Copy link
Collaborator

If you look at the requirements.txt you can see that our code currently runs with PyTorch 1.3 which in turn is compatible with CUDA 9.2 and 10.1.

@ks322181
Copy link
Author

Thanks i will check with higher version of cuda.

@ks322181 ks322181 closed this as completed Jan 6, 2020
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

3 participants