From a5e0912a27dc3b306cde5860e6c5abca062caab3 Mon Sep 17 00:00:00 2001
From: Thiago Crepaldi <thiago.crepaldi@microsoft.com>
Date: Thu, 22 Mar 2018 16:48:35 -0700
Subject: [PATCH] Fix automated cntk install script for windows

---
 Scripts/install/windows/ps/install.ps1 | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/Scripts/install/windows/ps/install.ps1 b/Scripts/install/windows/ps/install.ps1
index c4bb08503..d72544877 100644
--- a/Scripts/install/windows/ps/install.ps1
+++ b/Scripts/install/windows/ps/install.ps1
@@ -114,7 +114,10 @@ function WhlFileInfoFromVersionFile(
             throw "`nFatal Error: Malformed version information in [$versionFile]."
         }
         $cntkVersion = $cntkVersion -replace "-", "."
-        $cntkVersion = $cntkVersion -replace "^cntk\.", "cntk-"
+        if ($cntkTarget -eq "GPU")
+            $cntkVersion = $cntkVersion -replace "^cntk\.", "cntk_gpu-"
+        else:
+            $cntkVersion = $cntkVersion -replace "^cntk\.", "cntk-"
 
         return @{ Name = "{0}-cp{1}-cp{2}m-win_amd64.whl" -f $cntkVersion, $pyVersion, $pyVersion; CntkUrl = "{0}/{1}" -f $wheelBaseUrl, $cntkTarget }
     }