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

tensorflow-gpu==1.14.0下默认使用cpu进行训练 #13

Open
CavendishBosh opened this issue Nov 30, 2019 · 2 comments
Open

tensorflow-gpu==1.14.0下默认使用cpu进行训练 #13

CavendishBosh opened this issue Nov 30, 2019 · 2 comments

Comments

@CavendishBosh
Copy link

我train代码的时候tensorflow默认使用cpu,楼主大大这是啥情况啊?

@ursaminor0401
Copy link
Collaborator

我train代码的时候tensorflow默认使用cpu,楼主大大这是啥情况啊?

是不是CUDA和cuDNN版本问题,tensorflow 1.14.0 需要CUDA10.0

@CavendishBosh
Copy link
Author

是的,谢谢code主

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