Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR SHELL

pytorch cuda

evice = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
Source by pytorch.org #
 
PREVIOUS NEXT
Tagged: #pytorch #cuda
ADD COMMENT
Topic
Name
4+4 =