forked from microsoft/CNTK
/
DevInstall.ps1
183 lines (140 loc) · 7.67 KB
/
DevInstall.ps1
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
#
# Copyright (c) Microsoft. All rights reserved.
# Licensed under the MIT license. See LICENSE.md file in the project root for full license information.
#
<#
.SYNOPSIS
Use this cmdlet to install a CNTK development environment on your machine.
A detailed description can be found here: https://docs.microsoft.com/en-us/cognitive-toolkit/Setup-CNTK-with-script-on-Windows
.DESCRIPTION
The script will download and install the files necessary to create a CNTK development environment on your system.
It will analyse your machine and will determine which components are required.
The required components will be downloaded into [c:\installCacheCntk] and installed from that location.
Repeated operation of this script will reuse already downloaded components.
Before you can run this machine you should have read the instructions at
https://docs.microsoft.com/en-us/cognitive-toolkit/Setup-CNTK-with-script-on-Windows
.PARAMETER Execute
You can set this switch to false to prevent devInstall from performing any physical changes to the machine.
.PARAMETER NoConfirm
If you supply this optional parameter, the install script will execute operations without asking for user confirmation.
.PARAMETER localCache
This optional parameter can be used to specify the directory downloaded components will be stored in
.PARAMETER InstallLocation
The directory the install tools/dependencies will be installed in, by default [c:\local]
.PARAMETER AnacondaBasePath
This optional parameter allows you to specify the location of an Anaconda installation to be used or created on your
machine. If the directory exists on your machine, the script will continue under the assumption that this is a working
Anaconda 3 (4.1.1) (or compatible) installation, and will create the CNTK Python environment in that location.
By default a version of Anaconda3 will be installed into [C:\local\Anaconda3-4.1.1-Windows-x86_64]
.PARAMETER PyVersion
This is an optional parameter and can be used to specify the Python version used in the CNTK Python environment.
Supported values for this parameter are 27, 35, or 36. The default values is 35 (for a CNTK Python 35 environment).
.PARAMETER PyEnvironmentName
This optional parameter allows to specify the name of the environment that will be created during the installation process.
By default the environment will be named cntk-py<PyVersion>, where PyVersion is being replaced by the content of the <PyVersion>
parameter to this script. If this parameter is specified by you, no version substitution in the environment will be performed.
.PARAMETER NoPythonEnvironment
If this switch parameter is set, the install script will not create a CNTK Python environment during the installation process.
This allows creation of the desired environment after the installation.
.EXAMPLE
.\devInstall.ps1
Run the installer and install the development tools
.EXAMPLE
.\devInstall.ps1 -Execute:$false
Run the installer and see what operations would be performed, without actually performing these actions
.EXAMPLE
.\devInstall.ps1 -Execute -AnacondaBasePath d:\mytools\Anaconda34
If the directory [d:\mytools\Anaconda34] exists, the installer will assume it contains a complete Anaconda installation.
If the directory doesn't exist, Anaconda will be installed into this directory.
#>
[CmdletBinding()]
Param(
[Parameter(Mandatory=$false)] [switch] $Execute = $true,
[Parameter(Mandatory=$false)] [string] $localCache = "c:\installCacheCntk",
[Parameter(Mandatory=$false)] [string] $InstallLocation = "c:\local",
[Parameter(Mandatory=$false)] [string] $AnacondaBasePath,
[Parameter(Mandatory=$false)] [switch] $NoConfirm,
[Parameter(Mandatory=$false, ParameterSetName = "PythonVersion")] [ValidateSet("27", "35", "36")] [string] $PyVersion = "35",
[Parameter(Mandatory=$false, ParameterSetName = "PythonVersion")] [string] $PyEnvironmentName = "",
[Parameter(Mandatory=$true, ParameterSetName = "PythonNoEnvironment")] [switch] $NoPythonEnvironment)
$roboCopyCmd = "robocopy.exe"
if (-not $AnacondaBasePath) {
$AnacondaBasePath = Join-Path $InstallLocation Anaconda3-4.1.1-Windows-x86_64
}
#just make sure the supplied parameter don't end on a backslash
$localCache = (Join-Path $localCache .) | Split-Path
$InstallLocation = (Join-Path $InstallLocation .) | Split-Path
$AnacondaBasePath = (Join-Path $AnacondaBasePath .) | Split-Path
$localDir = $InstallLocation
# Get the current script's directory
$MyDir = Split-Path $MyInvocation.MyCommand.Definition
$CloneDirectory = Split-Path $mydir
$CloneDirectory = Split-Path $CloneDirectory
$CloneDirectory = Split-Path $CloneDirectory
$reponame = Split-Path $CloneDirectory -Leaf
$repositoryRootDir = Split-Path $CloneDirectory
$solutionfile = Join-Path $CloneDirectory "CNTK.SLN"
if (-not (Test-Path -Path $solutionFile -PathType Leaf)) {
Write-Warning "The install script was started out of the [$mydir] location. Based on this"
Write-Warning "[$CloneDirectory] should be the location of the CNTK sourcecode directory."
Write-Warning "The specified directory is not a valid clone of the CNTK Github project."
throw "Terminating install operation"
}
. "$MyDir\helper\Display"
. "$MyDir\helper\Common"
. "$MyDir\helper\Operations"
. "$MyDir\helper\Verification"
. "$MyDir\helper\Download"
. "$MyDir\helper\Action"
. "$MyDir\helper\PreRequisites"
Function main
{
try { if (-not (DisplayStart -NoConfirm $NoConfirm)) {
Write-Host
Write-Host " ... Quitting ... "
Write-Host
return
}
if (-not (Test-Path -Path $localCache)) {
new-item -Path $localcache -ItemType Container -ErrorAction Stop | Out-Null
}
ClearScriptVariables
$operation = @();
$operation += OpScanProgram
$operation += OpCheckVS2017
$operation += OpCheckCuda9
$operation += OpNVidiaCudnn7090 -cache $localCache -targetFolder $localDir
$operation += OpNvidiaCub174 -cache $localCache -targetFolder $localDir
$operation += OpCMake362 -cache $localCache
$operation += OpMSMPI70 -cache $localCache
$operation += OpMSMPI70SDK -cache $localCache
$operation += OpBoost160VS15 -cache $localCache -targetFolder $localDir
$operation += OpMKLDNN012 -cache $localCache -targetFolder $localDir
$operation += OpSwig3010 -cache $localCache -targetFolder $localDir
$operation += OpProtoBuf310VS17 -cache $localCache -targetFolder $localDir -repoDirectory $CloneDirectory
$operation += OpProtoBuf310VS17Prebuild -cache $localCache -targetFolder $localDir
$operation += OpZlibVS17 -cache $localCache -targetFolder $localDir -repoDirectory $CloneDirectory
$operation += OpZlibVS17Prebuild -cache $localCache -targetFolder $localDir
$operation += OpOpenCV31 -cache $localCache -targetFolder $localDir
$operation += OpAnaconda3411 -cache $localCache -AnacondaBasePath $AnacondaBasePath
if (-not $NoPythonEnvironment) {
$operation += OpAnacondaEnv -AnacondaBasePath $AnacondaBasePath -repoDir $repositoryRootDir -repoName $reponame -environmentName $PyEnvironmentName -pyVersion $PyVersion
}
$operationList = @()
$operationList += (VerifyOperations $operation)
PreReqOperations $operationList
if (DisplayAfterVerify -NoConfirm $NoConfirm -list $operationList) {
DownloadOperations $operationList
ActionOperations $operationList
DisplayEnd
}
}
catch {
Write-Host "Exception caught - function main / failure"
Write-Host ($Error[0]).Exception
Write-Host
}
}
main
exit 0
# vim:set expandtab shiftwidth=4 tabstop=4: