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

Torch not compiled with CUDA enabled #13

Open

HMAJG opened this issue Jun 19, 2020 · 1 comment

HMAJG commented Jun 19, 2020

The solution of using CPU to train in Windows Environment

1、 train.py
Change line 84 to Cuba = False
2、 train.py
Change line 108 to net = net.to(device)

Owner

bubbliiiing commented Jun 22, 2020

直接设置cuda为False不能训练么

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