test_gpu.py 67 B

1234
  1. # 测试是否有gpu
  2. import torch
  3. print(torch.cuda.is_available())