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训练？ #90

Open

Tian14267 opened this issue Mar 25, 2021 · 0 comments

Tian14267 commented Mar 25, 2021

你好，我想请问下，怎么进行GPU训练? 我设置了“os.environ["CUDA_VISIBLE_DEVICES"] = "0" ”，但是代码还是在CPU上面运行。请问如何设置GPU运行？

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