Fix cuda utils

pytorch-labs · Oct 11, 2024 · 5b127b1 · 5b127b1
1 parent 3a8816f
commit 5b127b1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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",