From 5b127b13bd47911315c47b31cd8891db8ec56bc6 Mon Sep 17 00:00:00 2001
From: Xu Zhao <i@xuzhao.net>
Date: Thu, 10 Oct 2024 22:10:15 -0400
Subject: [PATCH] Fix cuda utils

---
 utils/cuda_utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/utils/cuda_utils.py b/utils/cuda_utils.py
index 844f18d5..02bfcdb4 100644
--- a/utils/cuda_utils.py
+++ b/utils/cuda_utils.py
@@ -192,6 +192,7 @@ def check_torch_nightly_version(force_date: Optional[str] = None):
 
 
 if __name__ == "__main__":
+    import argparse
     parser = argparse.ArgumentParser()
     parser.add_argument(
         "--cudaver",