New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

#

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Jump to bottom

PyTorch notebooks should detect GPU and run on it if it's available #435

Open

dniku opened this issue May 19, 2020 · 0 comments

Collaborator

dniku commented May 19, 2020

device = 'cuda' if torch.cuda.is_available() else 'cpu'
...
model = model.to(device)

# for free to join this conversation on GitHub. Already have an account? # to comment