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.

Sign up for GitHub

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

Jump to bottom

使用GPU训练需要修改哪些参数 #28

Open

YukeeWang opened this issue Mar 24, 2022 · 0 comments

YukeeWang commented Mar 24, 2022

感谢您的工作
我已经在电脑上成功运行了代码，但是貌似只在CPU上工作，我已经设置了gpu-options-allow-growth=True，且能看到有一部显存占用，但似乎训练都是再CPU上完成的，我需要调整别的什么配置么?多谢

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment