commit 6402a4278b25e0307a09ab7fda9df496bf268adb
author Will Feng <yf225@cornell.edu> Wed Jun 20 09:41:50 2018 -0400
committer Edward Z. Yang <ezyang@mit.edu> Wed Jun 20 09:41:50 2018 -0400
tree 4f230a1102639e3a2e67ce1c94a3eadd0f1576b3
parent be3e3f2ec81620c32ab931a38873ca9fd1464046

Improve win-build.sh for local build (#8674)

.jenkins/pytorch/win-build.sh

diff --git a/.jenkins/pytorch/win-build.sh b/.jenkins/pytorch/win-build.sh
index c98f243..48a7145 100755
--- a/.jenkins/pytorch/win-build.sh
+++ b/.jenkins/pytorch/win-build.sh
@@ -1,6 +1,13 @@
 #!/bin/bash
 
 # If you want to rebuild, run this with REBUILD=1
+# If you want to build with CUDA, run this with USE_CUDA=1
+# If you want to build without CUDA, run this with USE_CUDA=0
+
+if [ ! -f setup.py ]; then
+  echo "ERROR: Please run this build script from PyTorch root directory."
+  exit 1
+fi
 
 COMPACT_JOB_NAME=pytorch-win-ws2016-cuda9-cudnn7-py3-build
 source "$(dirname "${BASH_SOURCE[0]}")/common.sh"
@@ -109,24 +116,35 @@
 
 set CMAKE_GENERATOR=Ninja
 
-if "%REBUILD%"=="" (
-  set NO_CUDA=1
-  python setup.py install
-)
-if errorlevel 1 exit /b 1
-if not errorlevel 0 exit /b 1
-if "%REBUILD%"=="" (
-  sccache --show-stats
-  sccache --zero-stats
-  rd /s /q C:\\Jenkins\\Miniconda3\\Lib\\site-packages\\torch
-  copy %CD%\\tmp_bin\\sccache.exe tmp_bin\\nvcc.exe
+if not "%USE_CUDA%"=="1" (
+  if "%REBUILD%"=="" (
+    set NO_CUDA=1
+    python setup.py install
+  )
+  if errorlevel 1 exit /b 1
+  if not errorlevel 0 exit /b 1
 )
 
-set CUDA_NVCC_EXECUTABLE=%CD%\\tmp_bin\\nvcc
+if not "%USE_CUDA%"=="0" (
+  if "%REBUILD%"=="" (
+    sccache --show-stats
+    sccache --zero-stats
+    rd /s /q C:\\Jenkins\\Miniconda3\\Lib\\site-packages\\torch
+    copy %CD%\\tmp_bin\\sccache.exe tmp_bin\\nvcc.exe
+  )
 
-if "%REBUILD%"=="" set NO_CUDA=
+  set CUDA_NVCC_EXECUTABLE=%CD%\\tmp_bin\\nvcc
 
-python setup.py install && sccache --show-stats && 7z a %IMAGE_COMMIT_TAG%.7z C:\\Jenkins\\Miniconda3\\Lib\\site-packages\\torch && python ci_scripts\\upload_image.py %IMAGE_COMMIT_TAG%.7z
+  if "%REBUILD%"=="" set NO_CUDA=0
+
+  python setup.py install && sccache --show-stats && (
+    if "%BUILD_ENVIRONMENT%"=="" (
+      echo "NOTE: To run `import torch`, please make sure to activate the conda environment by running `call C:\\Jenkins\\Miniconda3\\Scripts\\activate.bat C:\\Jenkins\\Miniconda3` in Command Prompt before running Git Bash."
+    ) else (
+      7z a %IMAGE_COMMIT_TAG%.7z C:\\Jenkins\\Miniconda3\\Lib\\site-packages\\torch && python ci_scripts\\upload_image.py %IMAGE_COMMIT_TAG%.7z
+    )
+  )
+)
 
 EOL