From 6b5b95d6a7fa474e0cc42aa9b8a015ac28679bd6 Mon Sep 17 00:00:00 2001
From: Leon <88978827@qq.com>
Date: Mon, 19 Feb 2024 15:50:20 +0800
Subject: [PATCH] update

---
 checker_torch.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)
 create mode 100644 checker_torch.py

diff --git a/checker_torch.py b/checker_torch.py
new file mode 100644
index 00000000..60fb4e01
--- /dev/null
+++ b/checker_torch.py
@@ -0,0 +1,19 @@
+"""
+@File: checker_torch.py
+@Author: Leon@spark2fire.cn
+@Date: 2/18/2024
+@QQ: 88978827
+"""
+import torch
+
+print('CUDA版本:', torch.version.cuda)
+print('Pytorch版本:', torch.__version__)
+print('显卡是否可用:', '可用' if (torch.cuda.is_available()) else '不可用')
+print('显卡数量:', torch.cuda.device_count())
+print('显卡是否支持BF16数字格式:', '支持' if (torch.cuda.is_bf16_supported()) else '不支持')
+print('当前显卡型号:', torch.cuda.get_device_name())
+print('当前显卡的CUDA算力:', torch.cuda.get_device_capability())
+print('当前显卡的总显存:', torch.cuda.get_device_properties(0).total_memory / 1024 / 1024 / 1024, 'GB')
+print('是否支持TensorCore:', '支持' if (torch.cuda.get_device_properties(0).major >= 7) else '不支持')
+print('当前显卡的显存使用率:', torch.cuda.memory_allocated(0) / torch.cuda.get_device_properties(0).total_memory * 100,
+      '%')