2023-DnB-Project/example/torch_check.py

3 lines
72 B
Python
Raw Permalink Normal View History

2023-06-26 16:32:02 +08:00
import torch
print(torch.__version__)
print(torch.cuda.is_available())