Add flag PYTORCH_BUILD_VERSION

Author: peterjc123

README.md

@@ -88,6 +88,10 @@ ImportError: DLL load failed: The specified procedure could not be found.
 You can download it and put it in the PyTorch directory or use it in a standalone way.
 There're more details about MSVC 2017 setup in [#23](https://github.com/peterjc123/pytorch-scripts/issues/23).
 ```Powershell
+# You can specify which version you want to build
+# If you omit it, it will build for the master branch on default
+set PYTORCH_BUILD_VERSION=0.4.1
+
 # If you don't want to override the default settings
 auto.bat
 

internal/clone.bat

@@ -1,4 +1,15 @@
 @echo off
 
 git clone --recursive https://github.com/pytorch/pytorch
+
+IF NOT "%PYTORCH_BUILD_VERSION%"=="" (
+    git checkout tags/v%PYTORCH_BUILD_VERSION%
+    IF ERRORLEVEL 1 git checkout v%PYTORCH_BUILD_VERSION%
+    IF ERRORLEVEL 1 (
+        echo Version %PYTORCH_BUILD_VERSION% not found, staying on the master branch
+    ) ELSE (
+        echo Building for version %PYTORCH_BUILD_VERSION%
+    )
+)
+
 cd pytorch

internal/setup.bat

@@ -7,6 +7,7 @@ if "%NO_CUDA%"==""  echo CUDA_PATH=%CUDA_PATH%
 if NOT "%CC%"==""   echo CC=%CC%
 if NOT "%CXX%"==""  echo CXX=%CXX%
 if NOT "%DISTUTILS_USE_SDK%"==""  echo DISTUTILS_USE_SDK=%DISTUTILS_USE_SDK%
+IF NOT "%PYTORCH_BUILD_VERSION%"=="" echo PYTORCH_BUILD_VERSION=%PYTORCH_BUILD_VERSION%
 
 setlocal EnableDelayedExpansion
 IF /I "%~1" NEQ "/y" (