Google Colab KeyError: 'COLAB_TPU_ADDR'












1















I'm trying to run a simple MNIST classifier on Google Colab using the TPU option. After creating the model using Keras, I am trying to convert it into TPU by:



import tensorflow as tf
import os

tpu_model = tf.contrib.tpu.keras_to_tpu_model(
model,
strategy=tf.contrib.tpu.TPUDistributionStrategy(
tf.contrib.cluster_resolver.TPUClusterResolver(tpu='grpc://' + os.environ['COLAB_TPU_ADDR'])
)
)
tpu_model.compile(
optimizer='rmsprop',
loss='categorical_crossentropy',
metrics=['accuracy']
)


print(model.summary())


And the error I'm getting is:



---------------------------------------------------------------------------
KeyError Traceback (most recent call last)
<ipython-input-5-63c528142aab> in <module>()
5 model,
6 strategy=tf.contrib.tpu.TPUDistributionStrategy(
----> 7 tf.contrib.cluster_resolver.TPUClusterResolver(tpu='grpc://' + os.environ['COLAB_TPU_ADDR'])
8 )
9 )

/usr/lib/python3.6/os.py in __getitem__(self, key)
667 except KeyError:
668 # raise KeyError with the original key value
--> 669 raise KeyError(key) from None
670 return self.decodevalue(value)
671

KeyError: 'COLAB_TPU_ADDR'


It looks like I need to change the TPU address but been googling and haven't found anything yet. Appreciate some help, thanks!










share|improve this question





























    1















    I'm trying to run a simple MNIST classifier on Google Colab using the TPU option. After creating the model using Keras, I am trying to convert it into TPU by:



    import tensorflow as tf
    import os

    tpu_model = tf.contrib.tpu.keras_to_tpu_model(
    model,
    strategy=tf.contrib.tpu.TPUDistributionStrategy(
    tf.contrib.cluster_resolver.TPUClusterResolver(tpu='grpc://' + os.environ['COLAB_TPU_ADDR'])
    )
    )
    tpu_model.compile(
    optimizer='rmsprop',
    loss='categorical_crossentropy',
    metrics=['accuracy']
    )


    print(model.summary())


    And the error I'm getting is:



    ---------------------------------------------------------------------------
    KeyError Traceback (most recent call last)
    <ipython-input-5-63c528142aab> in <module>()
    5 model,
    6 strategy=tf.contrib.tpu.TPUDistributionStrategy(
    ----> 7 tf.contrib.cluster_resolver.TPUClusterResolver(tpu='grpc://' + os.environ['COLAB_TPU_ADDR'])
    8 )
    9 )

    /usr/lib/python3.6/os.py in __getitem__(self, key)
    667 except KeyError:
    668 # raise KeyError with the original key value
    --> 669 raise KeyError(key) from None
    670 return self.decodevalue(value)
    671

    KeyError: 'COLAB_TPU_ADDR'


    It looks like I need to change the TPU address but been googling and haven't found anything yet. Appreciate some help, thanks!










    share|improve this question



























      1












      1








      1


      0






      I'm trying to run a simple MNIST classifier on Google Colab using the TPU option. After creating the model using Keras, I am trying to convert it into TPU by:



      import tensorflow as tf
      import os

      tpu_model = tf.contrib.tpu.keras_to_tpu_model(
      model,
      strategy=tf.contrib.tpu.TPUDistributionStrategy(
      tf.contrib.cluster_resolver.TPUClusterResolver(tpu='grpc://' + os.environ['COLAB_TPU_ADDR'])
      )
      )
      tpu_model.compile(
      optimizer='rmsprop',
      loss='categorical_crossentropy',
      metrics=['accuracy']
      )


      print(model.summary())


      And the error I'm getting is:



      ---------------------------------------------------------------------------
      KeyError Traceback (most recent call last)
      <ipython-input-5-63c528142aab> in <module>()
      5 model,
      6 strategy=tf.contrib.tpu.TPUDistributionStrategy(
      ----> 7 tf.contrib.cluster_resolver.TPUClusterResolver(tpu='grpc://' + os.environ['COLAB_TPU_ADDR'])
      8 )
      9 )

      /usr/lib/python3.6/os.py in __getitem__(self, key)
      667 except KeyError:
      668 # raise KeyError with the original key value
      --> 669 raise KeyError(key) from None
      670 return self.decodevalue(value)
      671

      KeyError: 'COLAB_TPU_ADDR'


      It looks like I need to change the TPU address but been googling and haven't found anything yet. Appreciate some help, thanks!










      share|improve this question
















      I'm trying to run a simple MNIST classifier on Google Colab using the TPU option. After creating the model using Keras, I am trying to convert it into TPU by:



      import tensorflow as tf
      import os

      tpu_model = tf.contrib.tpu.keras_to_tpu_model(
      model,
      strategy=tf.contrib.tpu.TPUDistributionStrategy(
      tf.contrib.cluster_resolver.TPUClusterResolver(tpu='grpc://' + os.environ['COLAB_TPU_ADDR'])
      )
      )
      tpu_model.compile(
      optimizer='rmsprop',
      loss='categorical_crossentropy',
      metrics=['accuracy']
      )


      print(model.summary())


      And the error I'm getting is:



      ---------------------------------------------------------------------------
      KeyError Traceback (most recent call last)
      <ipython-input-5-63c528142aab> in <module>()
      5 model,
      6 strategy=tf.contrib.tpu.TPUDistributionStrategy(
      ----> 7 tf.contrib.cluster_resolver.TPUClusterResolver(tpu='grpc://' + os.environ['COLAB_TPU_ADDR'])
      8 )
      9 )

      /usr/lib/python3.6/os.py in __getitem__(self, key)
      667 except KeyError:
      668 # raise KeyError with the original key value
      --> 669 raise KeyError(key) from None
      670 return self.decodevalue(value)
      671

      KeyError: 'COLAB_TPU_ADDR'


      It looks like I need to change the TPU address but been googling and haven't found anything yet. Appreciate some help, thanks!







      deep-learning google-colaboratory google-cloud-tpu






      share|improve this question















      share|improve this question













      share|improve this question




      share|improve this question








      edited Nov 20 '18 at 0:29









      Bob Smith

      6,58212129




      6,58212129










      asked Nov 4 '18 at 15:43









      AntoAnto

      235




      235
























          1 Answer
          1






          active

          oldest

          votes


















          3














          You'll need to change the backend to include a TPU using the notebook settings available in the Edit -> Notebook settings menu.



          enter image description here






          share|improve this answer























            Your Answer






            StackExchange.ifUsing("editor", function () {
            StackExchange.using("externalEditor", function () {
            StackExchange.using("snippets", function () {
            StackExchange.snippets.init();
            });
            });
            }, "code-snippets");

            StackExchange.ready(function() {
            var channelOptions = {
            tags: "".split(" "),
            id: "1"
            };
            initTagRenderer("".split(" "), "".split(" "), channelOptions);

            StackExchange.using("externalEditor", function() {
            // Have to fire editor after snippets, if snippets enabled
            if (StackExchange.settings.snippets.snippetsEnabled) {
            StackExchange.using("snippets", function() {
            createEditor();
            });
            }
            else {
            createEditor();
            }
            });

            function createEditor() {
            StackExchange.prepareEditor({
            heartbeatType: 'answer',
            autoActivateHeartbeat: false,
            convertImagesToLinks: true,
            noModals: true,
            showLowRepImageUploadWarning: true,
            reputationToPostImages: 10,
            bindNavPrevention: true,
            postfix: "",
            imageUploader: {
            brandingHtml: "Powered by u003ca class="icon-imgur-white" href="https://imgur.com/"u003eu003c/au003e",
            contentPolicyHtml: "User contributions licensed under u003ca href="https://creativecommons.org/licenses/by-sa/3.0/"u003ecc by-sa 3.0 with attribution requiredu003c/au003e u003ca href="https://stackoverflow.com/legal/content-policy"u003e(content policy)u003c/au003e",
            allowUrls: true
            },
            onDemand: true,
            discardSelector: ".discard-answer"
            ,immediatelyShowMarkdownHelp:true
            });


            }
            });














            draft saved

            draft discarded


















            StackExchange.ready(
            function () {
            StackExchange.openid.initPostLogin('.new-post-login', 'https%3a%2f%2fstackoverflow.com%2fquestions%2f53142474%2fgoogle-colab-keyerror-colab-tpu-addr%23new-answer', 'question_page');
            }
            );

            Post as a guest















            Required, but never shown

























            1 Answer
            1






            active

            oldest

            votes








            1 Answer
            1






            active

            oldest

            votes









            active

            oldest

            votes






            active

            oldest

            votes









            3














            You'll need to change the backend to include a TPU using the notebook settings available in the Edit -> Notebook settings menu.



            enter image description here






            share|improve this answer




























              3














              You'll need to change the backend to include a TPU using the notebook settings available in the Edit -> Notebook settings menu.



              enter image description here






              share|improve this answer


























                3












                3








                3







                You'll need to change the backend to include a TPU using the notebook settings available in the Edit -> Notebook settings menu.



                enter image description here






                share|improve this answer













                You'll need to change the backend to include a TPU using the notebook settings available in the Edit -> Notebook settings menu.



                enter image description here







                share|improve this answer












                share|improve this answer



                share|improve this answer










                answered Nov 4 '18 at 16:55









                Bob SmithBob Smith

                6,58212129




                6,58212129






























                    draft saved

                    draft discarded




















































                    Thanks for contributing an answer to Stack Overflow!


                    • Please be sure to answer the question. Provide details and share your research!

                    But avoid



                    • Asking for help, clarification, or responding to other answers.

                    • Making statements based on opinion; back them up with references or personal experience.


                    To learn more, see our tips on writing great answers.




                    draft saved


                    draft discarded














                    StackExchange.ready(
                    function () {
                    StackExchange.openid.initPostLogin('.new-post-login', 'https%3a%2f%2fstackoverflow.com%2fquestions%2f53142474%2fgoogle-colab-keyerror-colab-tpu-addr%23new-answer', 'question_page');
                    }
                    );

                    Post as a guest















                    Required, but never shown





















































                    Required, but never shown














                    Required, but never shown












                    Required, but never shown







                    Required, but never shown

































                    Required, but never shown














                    Required, but never shown












                    Required, but never shown







                    Required, but never shown







                    Popular posts from this blog

                    android studio warns about leanback feature tag usage required on manifest while using Unity exported app?

                    SQL update select statement

                    'app-layout' is not a known element: how to share Component with different Modules