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似乎没有被使用 #21

Open

Huhu-Adora opened this issue Jun 29, 2019 · 3 comments

Huhu-Adora commented Jun 29, 2019

训练时cpu占用100%而gpu占用仅1%

stefwoo commented Aug 8, 2019

需要自行修改代码，
模型和数据后面加上.cuda()方法即可，
速度确实可以快很多。
具体可google

pprp commented Oct 23, 2019

https://github.com/pprp/captcha_identify.torch 基于这个库进行改进，支持GPU CPU

jiangyv1718 commented Jul 27, 2023

模型和数据.to('cuda:0')就行

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