Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR PYTHON

pytorch get gpu number

torch.cuda.device_count()
Source by discuss.pytorch.org #
 
PREVIOUS NEXT
Tagged: #pytorch #gpu #number
ADD COMMENT
Topic
Name
2+2 =