fix exp2 pytorch rewrite fatal train issue

exp2/modified/sketchpad.py

@@ -2,11 +2,10 @@ from pathlib import Path
 import sys
 import typing
 import tkinter as tk
-from tkinter import messagebox
 from predict import PredictResult, Predictor
 
 sys.path.append(str(Path(__file__).resolve().parent.parent.parent))
-import gpu_utils
+import pytorch_gpu_utils
 
 
 class SketchpadApp:
@@ -182,7 +181,8 @@ class SketchpadApp:
     # endregion
 
 if __name__ == "__main__":
-    gpu_utils.print_gpu_availability()
+    pytorch_gpu_utils.print_gpu_availability()
+    
     predictor = Predictor()
 
     root = tk.Tk()