Fix automated cntk install script for windows

This commit is contained in:
Thiago Crepaldi 2018-03-22 16:48:35 -07:00
Родитель 5b04f46aa4
Коммит a5e0912a27
1 изменённых файлов: 4 добавлений и 1 удалений

Просмотреть файл

@ -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 }
}