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

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

Open

chenxs1427 opened this issue Nov 19, 2020 · 2 comments

chenxs1427 commented Nov 19, 2020

为啥不默认GPU的

Thresh-Frazier commented Feb 23, 2021

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

2212240289 commented Apr 6, 2022

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

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