Skip to content

Commit 11c82c8

Browse files
author
Thiago Crepaldi
committed
Fix install script for windows
1 parent cb28c0c commit 11c82c8

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

Scripts/install/windows/ps/install.ps1

+5-1
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,11 @@ function WhlFileInfoFromVersionFile(
114114
throw "`nFatal Error: Malformed version information in [$versionFile]."
115115
}
116116
$cntkVersion = $cntkVersion -replace "-", "."
117-
$cntkVersion = $cntkVersion -replace "^cntk\.", "cntk-"
117+
if ($cntkTarget -eq "GPU") {
118+
$cntkVersion = $cntkVersion -replace "^cntk\.", "cntk_gpu-"
119+
} else {
120+
$cntkVersion = $cntkVersion -replace "^cntk\.", "cntk-"
121+
}
118122

119123
return @{ Name = "{0}-cp{1}-cp{2}m-win_amd64.whl" -f $cntkVersion, $pyVersion, $pyVersion; CntkUrl = "{0}/{1}" -f $wheelBaseUrl, $cntkTarget }
120124
}

0 commit comments

Comments
 (0)