Skip to content

Commit 99a5f7b

Browse files
authored
Rename --cloud-gpu tesla to v100 (#540)
* Add --cloud-gpu v100 option to cml-runner * Add deprecation message for "tesla" GPU
1 parent a86306d commit 99a5f7b

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

bin/cml-runner.js

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -146,6 +146,10 @@ const run_cloud = async (opts) => {
146146
if (tf_file) {
147147
tpl = await fs.writeFile(tf_main_path, await fs.readFile(tf_file));
148148
} else {
149+
if (gpu === 'tesla')
150+
console.log(
151+
'GPU model "tesla" has been deprecated; please use "v100" instead.'
152+
);
149153
tpl = tf.iterative_cml_runner_tpl({
150154
repo,
151155
token,
@@ -157,7 +161,7 @@ const run_cloud = async (opts) => {
157161
cloud,
158162
region,
159163
type,
160-
gpu,
164+
gpu: gpu.replace(/^tesla$/, 'v100'),
161165
hdd_size,
162166
ssh_private,
163167
spot,
@@ -366,7 +370,7 @@ const opts = decamelize(
366370
)
367371
.default('cloud-gpu')
368372
.describe('cloud-gpu', 'GPU type.')
369-
.choices('cloud-gpu', ['nogpu', 'k80', 'tesla'])
373+
.choices('cloud-gpu', ['nogpu', 'k80', 'v100', 'tesla'])
370374
.coerce('cloud-gpu-type', (val) => (val === 'nogpu' ? null : val))
371375
.default('cloud-hdd-size')
372376
.describe('cloud-hdd-size', 'HDD size in GB')

0 commit comments

Comments
 (0)