Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR PYTHON

pytorch check if cuda is available

import torch
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
PREVIOUS NEXT
Tagged: #pytorch #check #cuda
ADD COMMENT
Topic
Name
8+1 =