TensorFlow/torch-test.py

5 lines
49 B
Python
Raw Normal View History

2021-03-24 21:37:44 +00:00
import torch
print(torch.cuda.is_available())