Skip to content
This repository was archived by the owner on Jul 7, 2023. It is now read-only.

Commit aaabb96

Browse files
lgeigerafrozenator
authored andcommitted
Upgrade Cloud ML version to [email protected] (#1344)
tensor2tensor requires [`tensorflow>=1.12.0`](https://github.com/tensorflow/tensor2tensor/blob/master/setup.py#L59). This PR upgrades the version on Cloud ML to 1.12 in order to fix run time errors. Note: I only tested this change on Cloud ML *without* a TPU.
1 parent 7b6ef71 commit aaabb96

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tensor2tensor/utils/cloud_mlengine.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,14 +34,14 @@
3434
FLAGS = tf.flags.FLAGS
3535

3636
CONSOLE_URL = "https://console.cloud.google.com/mlengine/jobs/"
37-
RUNTIME_VERSION = "1.9"
37+
RUNTIME_VERSION = "1.12"
3838

3939

4040
class Gcloud(object):
4141
"""gcloud command strings."""
4242
# Note these can be modified by set_versions
43-
VM_VERSION = "tf-1-9"
44-
TPU_VERSION = "1.9"
43+
VM_VERSION = "tf-1-12"
44+
TPU_VERSION = "1.12"
4545

4646
@classmethod
4747
def set_versions(cls, vm, tpu):

0 commit comments

Comments
 (0)