fix small bug

c0c9073d · yan.yan · fe03ed93 · c0c9073d
Commit c0c9073d authored Oct 20, 2021 by yan.yan
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

setup.py setup.py +1 -0

No files found.
--- a/setup.py
+++ b/setup.py
@@ -31,6 +31,7 @@ if not cuda_ver:
    nvcc_version_str = nvcc_version.split("\n")[3]
    version_str: str = re.findall(r"release (\d+.\d+)",
                                    nvcc_version_str)[0]
+    cuda_ver = version_str
 cuda_ver = cuda_ver.replace(".", "") # 10.2 to 102

 RELEASE_NAME += "-cu{}".format(cuda_ver)