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