github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[setup] add local version label (#890)

pull/895/head
ver217 2022-04-27 15:26:12 +08:00 committed by GitHub
parent 1d0aba4153
commit daf59ff72e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
setup.py
View File

@ -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: