TensorFlow/torch-test.py

5 lines
49 B
Python

import torch
print(torch.cuda.is_available())