From 150b1a745350a3f9c7a713dd82907240beacef74 Mon Sep 17 00:00:00 2001 From: ver217 Date: Thu, 5 May 2022 14:59:12 +0800 Subject: [PATCH] update local version format (#909) --- setup.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index d995ef4c4..c3913302f 100644 --- a/setup.py +++ b/setup.py @@ -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