Fix install script for windows

microsoft · Apr 13, 2018 · 11c82c8 · 11c82c8
1 parent cb28c0c
commit 11c82c8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/Scripts/install/windows/ps/install.ps1 b/Scripts/install/windows/ps/install.ps1
@@ -114,7 +114,11 @@ 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 }
     }