Skip to content

torch.cuda.is_available() error #305

Answered by nazarPuriy
ankitmazumder asked this question in Q&A
Discussion options

You must be logged in to vote

You misspelled available.

Try:

import torch
device = "cuda" if torch.cuda.is_available() else "cpu"

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by mrdbourke
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants