Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR PYTHON

torch device

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
PREVIOUS NEXT
Tagged: #torch #device
ADD COMMENT
Topic
Name
6+6 =