Added nv_win_rgs.ps1 and setup_win_rgs.sh to scripts directory.

This commit is contained in:
Cormac Garvey 2019-07-31 09:42:53 -06:00
Родитель a34c3f4dd6
Коммит abbffcbcee
2 изменённых файлов: 81 добавлений и 0 удалений

16
scripts/nv_win_rgs.ps1 Normal file
Просмотреть файл

@ -0,0 +1,16 @@
#install RGS
param (
[Parameter(Mandatory=$true)][string] $lic_url,
[Parameter(Mandatory=$true)][string] $sw_url
)
$ProgressPreference = 'SilentlyContinue'
Write-Output $lic_url
Write-Output $sw_url
Set-Location C:\Windows\Temp
Invoke-WebRequest -OutFile C:\Windows\rgs_trial.lic $lic_url
Invoke-WebRequest -OutFile SenderSetup64.exe $sw_url
.\SenderSetup64.exe /z"/autoinstall /agreetolicense /clipboard /rgslicensefile=C:\Windows\rgs_trial.lic /noreboot" -Wait

65
scripts/setup_win_rgs.sh Executable file
Просмотреть файл

@ -0,0 +1,65 @@
#!/bin/bash
# Setup HP RGS on a Win10 VM
resource_group=$1
node_name=$2
nfs_server=$3
storage_accnt=$4
container="apps"
lic_file="hp-rgs/trial.lic"
rgs_exe="hp-rgs/SenderSetup64.exe"
start=$(date --utc -d "-2 hours" +%Y-%m-%dT%H:%M:%SZ)
end=$(date --utc -d "+1 year" +%Y-%m-%dT%H:%M:%SZ)
echo "Creating the SAS key"
rgs_sas=$(az storage container generate-sas --account-name $storage_accnt --name $container --permissions r --output tsv --start $start --expiry $end )
#echo $rgs_sas
hp_rgs_lic_url="https://${storage_accnt}.blob.core.windows.net/${container}/${lic_file}"
hp_rgs_sw_url="https://${storage_accnt}.blob.core.windows.net/${container}/${rgs_exe}"
hp_rgs_lic_url="${hp_rgs_lic_url}?${rgs_sas}"
hp_rgs_sw_url="${hp_rgs_sw_url}?${rgs_sas}"
echo "Applying GPU extensions..."
az vm extension set \
--resource-group $resource_group \
--vm-name $node_name \
--name NvidiaGpuDriverWindows \
--publisher Microsoft.HpcCompute \
--version 1.2 \
--settings '{}' \
--output table
echo "GPU extensions applied"
echo "Calling nfsclient.ps1..."
az vm run-command invoke \
--name $node_name \
--resource-group $resource_group \
--command-id RunPowerShellScript \
--scripts @$azhpc_dir/scripts/nfsclient.ps1 \
--parameters nfs_server=$nfs_server \
--output table
echo "NFS done"
echo "Installing RGS..."
az vm run-command invoke \
--name $node_name \
--resource-group $resource_group \
--command-id RunPowerShellScript \
--scripts @nv_win_rgs.ps1 \
--parameters lic_url="\"$hp_rgs_lic_url\"" sw_url="\"$hp_rgs_sw_url\"" \
--output table --debug
echo "RGS Installed"
az network nsg rule create \
--resource-group $resource_group \
--nsg-name ${node_name}NSG \
--name HP-RGS \
--priority 310 \
--destination-port-ranges 42966 \
--access Allow \
--description "Allow HP RGS" \
--output table