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

update local version format (#909)

pull/912/head
ver217 2022-05-05 14:59:12 +08:00 committed by GitHub
parent 3b1f5f07ce
commit 150b1a7453
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
setup.py
View File

@ -96,8 +96,8 @@ def get_version():
with open('version.txt') as f: with open('version.txt') as f:
version = f.read().strip() version = f.read().strip()
if build_cuda_ext: if build_cuda_ext:
torch_version = ''.join(torch.__version__.split('.')[:2]) torch_version = '.'.join(torch.__version__.split('.')[:2])
cuda_version = ''.join(get_cuda_bare_metal_version(CUDA_HOME)[1:]) cuda_version = '.'.join(get_cuda_bare_metal_version(CUDA_HOME)[1:])
version += f'+torch{torch_version}cu{cuda_version}' version += f'+torch{torch_version}cu{cuda_version}'
return version return version