codefuse-chatbot/tests/torch_test.py

3 lines
70 B
Python
Raw Permalink Normal View History

2023-09-28 10:58:58 +08:00
import torch
print(torch.__version__)
print(torch.cuda.is_available())