github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-04 18:40:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

[setup] add local version label (#890)

Browse Source
This commit is contained in:
ver217
2022-04-27 15:26:12 +08:00
committed by GitHub
parent 1d0aba4153
commit daf59ff72e
1 changed files with 6 additions and 1 deletions
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():
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: