fix exp2 pytorch rewrite fatal train issue

2025-11-30 22:01:56 +08:00
parent 48fcdfcc80
commit 43b807679f
13 changed files with 738 additions and 112 deletions
--- a/gpu_utils.py
+++ b/gpu_utils.py
@@ -1,17 +0,0 @@
-import torch
-
-
-def print_gpu_availability():
-    """打印PyTorch的GPU可用性"""
-    if torch.cuda.is_available():
-        print(f"GPU可用：{torch.cuda.get_device_name(0)}")
-    else:
-        print("GPU不可用")
-
-
-def get_gpu_device() -> torch.device:
-    """获取PyTorch的GPU设备"""
-    if torch.cuda.is_available():
-        return torch.device("cuda")
-    else:
-        raise Exception("找不到CUDA！")