Skip to content

GitLab

Unverified Commit daf59ff7 authored by ver217's avatar ver217 Committed by GitHub
Browse files

[setup] add local version label (#890)

parent 1d0aba41
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 1 deletion
setup.py
View file @ daf59ff7
...@@ -94,7 +94,12 @@ def fetch_readme(): ...@@ -94,7 +94,12 @@ def fetch_readme():
def get_version(): def get_version():
with open('version.txt') as f: with open('version.txt') as f:
return f.read().strip() version = f.read().strip()
if build_cuda_ext:
torch_version = ''.join(torch.__version__.split('.')[:2])
cuda_version = ''.join(get_cuda_bare_metal_version(CUDA_HOME)[1:])
version += f'+torch{torch_version}cu{cuda_version}'
return version
if build_cuda_ext: if build_cuda_ext:
......
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