Configure GPU type/count from command line in GCE scripts (#1376)

* Configure GPU type/count from command line in GCE scripts

* Change CLI to input full leader machine type information with GPU
This commit is contained in:
Pankaj Garg 2018-09-27 11:55:56 -07:00 committed by GitHub
parent dbc1ffc75e
commit 7fb7839c8f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 1 deletions

View File

@ -71,6 +71,7 @@ Manage testnet instances
-P - Use public network IP addresses (default: $publicNetwork)
-z [zone] - Zone for the nodes (default: $zone)
-g - Enable GPU (default: $enableGpu)
-G - Enable GPU, and set count/type of GPUs to use (e.g $cpuLeaderMachineType --accelerator count=4,type=nvidia-tesla-k80)
-a [address] - Set the leader node's external IP address to this value.
For GCE, [address] is the "name" of the desired External
IP Address.
@ -94,7 +95,7 @@ command=$1
shift
[[ $command = create || $command = config || $command = delete ]] || usage "Invalid command: $command"
while getopts "h?p:Pn:c:z:ga:d:" opt; do
while getopts "h?p:Pn:c:z:gG:a:d:" opt; do
case $opt in
h | \?)
usage
@ -119,6 +120,10 @@ while getopts "h?p:Pn:c:z:ga:d:" opt; do
enableGpu=true
leaderMachineType="$gpuLeaderMachineType"
;;
G)
enableGpu=true
leaderMachineType="$OPTARG"
;;
a)
leaderAddress=$OPTARG
;;