Skip to content

GitLab

Commit c0c9073d authored by yan.yan's avatar yan.yan
Browse files

fix small bug

parent fe03ed93
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
setup.py
View file @ c0c9073d
...@@ -31,6 +31,7 @@ if not cuda_ver: ...@@ -31,6 +31,7 @@ if not cuda_ver:
nvcc_version_str = nvcc_version.split("\n")[3] nvcc_version_str = nvcc_version.split("\n")[3]
version_str: str = re.findall(r"release (\d+.\d+)", version_str: str = re.findall(r"release (\d+.\d+)",
nvcc_version_str)[0] nvcc_version_str)[0]
cuda_ver = version_str
cuda_ver = cuda_ver.replace(".", "") # 10.2 to 102 cuda_ver = cuda_ver.replace(".", "") # 10.2 to 102
RELEASE_NAME += "-cu{}".format(cuda_ver) RELEASE_NAME += "-cu{}".format(cuda_ver)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment