pytorch-captcha-recognition 直接就用cpu训练，把我的电脑cpu给爆了 >

直接就用cpu训练，把我的电脑cpu给爆了 >_<

Open chenxs1427 opened this issue 5 years ago • 2 comments

为啥不默认GPU的

Nov 19 '20 07:11 chenxs1427

尝试下在train.py中修改： device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

Feb 23 '21 01:02 Thresh-Frazier

试下在train.py中修改： device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") 这行代码放在哪里呀，没学过这些

Apr 06 '22 15:04 2212240289