From 6567b8f018507486f67ea56943e94bb8d41e93a4 Mon Sep 17 00:00:00 2001
From: peter <peter_jiachen@163.com>
Date: Sun, 22 Sep 2019 14:22:55 +0800
Subject: [PATCH] Fix nightly wheels builds for Windows

---
 packaging/windows/build_vision.bat | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/packaging/windows/build_vision.bat b/packaging/windows/build_vision.bat
index 162dba44917..995c43905cb 100644
--- a/packaging/windows/build_vision.bat
+++ b/packaging/windows/build_vision.bat
@@ -117,6 +117,16 @@ for %%v in (%DESIRED_PYTHON_PREFIX%) do (
     :: Activate Python Environment
     set PYTHON_PREFIX=%%v
     set "PATH=%CONDA_HOME%\envs\%%v;%CONDA_HOME%\envs\%%v\scripts;%CONDA_HOME%\envs\%%v\Library\bin;%ORIG_PATH%"
+    if defined INCLUDE (
+        set "INCLUDE=%INCLUDE%;%CONDA_HOME%\envs\%%v\Library\include"
+    ) else (
+        set "INCLUDE=%CONDA_HOME%\envs\%%v\Library\include"
+    )
+    if defined LIB (
+        set "LIB=%LIB%;%CONDA_HOME%\envs\%%v\Library\lib"
+    ) else (
+        set "LIB=%CONDA_HOME%\envs\%%v\Library\lib"
+    )
     @setlocal
     :: Set Flags
     if NOT "%CUDA_VERSION%"=="cpu" (