Skip to content

GitLab

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

update local version format (#909)

parent 3b1f5f07
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
setup.py
View file @ 150b1a74
......@@ -96,8 +96,8 @@ def get_version():
with open('version.txt') as f:
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:])
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
......
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