pytorch-captcha-recognition icon indicating copy to clipboard operation
pytorch-captcha-recognition copied to clipboard

Published 20 hours ago • verified publisher icon dee1024

gpu似乎没有被使用

Open Huhu-Adora opened this issue 6 years ago • 3 comments

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

Huhu-Adora avatar Jun 29 '19 05:06 Huhu-Adora

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

stefwoo avatar Aug 08 '19 10:08 stefwoo

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

pprp avatar Oct 23 '19 03:10 pprp

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

jiangyv1718 avatar Jul 27 '23 06:07 jiangyv1718