PYTORCH_BUILD_VERSION to setup.py

tgeral68 · Jan 18, 2017 · 57a2ccf · 57a2ccf
1 parent 205b9bc
commit 57a2ccf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -313,9 +313,9 @@ def make_relative_rpath(path):
     extensions.append(THCUNN)
 
 version="0.1"
-if os.getenv('PYTORCH_BINARY_VERSION'):
-    version = os.getenv('PYTORCH_BINARY_VERSION') \
-              + '_' + os.getenv('PYTORCH_BINARY_NUMBER')
+if os.getenv('PYTORCH_BUILD_VERSION'):
+    version = os.getenv('PYTORCH_BUILD_VERSION') \
+              + '_' + os.getenv('PYTORCH_BUILD_NUMBER')
 
 setup(name="torch", version=version,
     ext_modules=extensions,