cuda_test.py 47 B

123
  1. import torch
  2. print(torch.cuda.device_count())