add cuda warning on windows (ModelCloud#942)

* add cuda warning * Update setup.py --------- Co-authored-by: Qubitium-ModelCloud <[email protected]>
ZX-ModelCloud · Dec 20, 2024 · fa377d9 · fa377d9
1 parent 10ab008
commit fa377d9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -101,6 +101,9 @@ def get_version_tag(is_cuda_release: bool = True) -> str:
     if not got_cuda_v6:
         BUILD_CUDA_EXT = False
 
+        if sys.platform == "win32" and 'cu+' not in torch.__version__:
+            print("No CUDA device detected: avoid installing torch from PyPi which may not have bundle CUDA support for Windows.\nInstall via PyTorch: `https://pytorch.org/get-started/locally/`")
+
     # if cuda compute is < 8.0, always force build since we only compile cached wheels for >= 8.0
     if BUILD_CUDA_EXT and not FORCE_BUILD:
         if got_cuda_between_v6_and_v8: