-
Notifications
You must be signed in to change notification settings - Fork 195
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
Comments
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')) |
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. |
Can we close this issue? |
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. |
If you look at the |
Thanks i will check with higher version of cuda. |
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.
The text was updated successfully, but these errors were encountered: