Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR PYTHON

device gpu pytorch

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
Source by stackoverflow.com #
 
PREVIOUS NEXT
Tagged: #device #gpu #pytorch
ADD COMMENT
Topic
Name
8+6 =