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

torch.cuda.is_available() is False. #1

Closed
longmalongma opened this issue Jul 28, 2019 · 4 comments
Closed

torch.cuda.is_available() is False. #1

longmalongma opened this issue Jul 28, 2019 · 4 comments

Comments

@longmalongma
Copy link

Hello,Think your for your work,when I tried to run your code ,I met this problem.can you tell me how to solve this problem?
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='cpu' to map your storages to the CPU.

@Hxyou
Copy link
Collaborator

Hxyou commented Jul 30, 2019

It seems you need a GPU and correctly configure it with CUDA, CUDNN.

@longmalongma
Copy link
Author

Think your for your reply,I successfully make it ,I want to know your version about cuda,cudnn,pytorch,do you have code about this parper on the tensorflow?

@Hxyou
Copy link
Collaborator

Hxyou commented Jul 30, 2019

The version of cuda and pytorch are listed in readme of our project. And the version of cudnn is 7.1.2. I am sorry to not have tensorflow code.

@longmalongma
Copy link
Author

longmalongma commented Jul 30, 2019 via email

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

2 participants