Only check for 'bin' folder in cudnn path on windows.

microsoft · Nov 29, 2018 · 91f1866 · 91f1866
1 parent 2f61926
commit 91f1866
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/tools/ci_build/build.py b/tools/ci_build/build.py
@@ -296,12 +296,13 @@ def setup_cuda_vars(args):
                       .format(cuda_home, cuda_home_valid, cudnn_home, cudnn_home_valid))
             sys.exit(-1)
 
-        if (not os.path.exists(os.path.join(cudnn_home, "bin"))):
-            log.error("cudnn_home path should include the 'cuda' folder, and must contain the CUDNN 'bin' directory.")
-            log.error("cudnn_home='{}'".format(cudnn_home))
-            sys.exit(-1)
-
         if (is_windows()):
+            # Validate that the cudnn_home is pointing at the right level
+            if (not os.path.exists(os.path.join(cudnn_home, "bin"))):
+                log.error("cudnn_home path should include the 'cuda' folder, and must contain the CUDNN 'bin' directory.")
+                log.error("cudnn_home='{}'".format(cudnn_home))
+                sys.exit(-1)
+
             os.environ["CUDA_PATH"] = cuda_home
             os.environ["CUDA_TOOLKIT_ROOT_DIR"] = cuda_home