Hi,
I’m looking for an example  to learn how to use TPUs on Colab running PyTorch.
I’m glad to find the Simple NLP Example 6 which is unfortunately not working.
Running w/o modifications leads to following error message running the last cell:

from accelerate import notebook_launcher
notebook_launcher(training_function)

---------------------------------------------------------------------------

ImportError                               Traceback (most recent call last)

<ipython-input-50-a91f3c0bb4fd> in <module>()
      1 from accelerate import notebook_launcher
      2 
----> 3 notebook_launcher(training_function)

1 frames

/usr/local/lib/python3.7/dist-packages/torch_xla/__init__.py in <module>()
     99 from ._patched_functions import _apply_patches
    100 from .version import __version__
--> 101 import _XLAC
    102 
    103 

ImportError: /usr/local/lib/python3.7/dist-packages/_XLAC.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZNK3c1010TensorImpl20is_contiguous_customENS_12MemoryFormatE


---------------------------------------------------------------------------
NOTE: If your import is failing due to a missing package, you can
manually install dependencies using either !pip or !apt.

To view examples of installing some common dependencies, click the
"Open Examples" button below.
---------------------------------------------------------------------------

I found a workaround description here 1 which says:
... downgrading PyTorch to torch-1.8.2+cpu,
but that leads to another error message
ProcessExitedException: process 0 terminated with signal SIGSEGV

What is necessary to run that example?
Do you know any other example that meets my requirements (Colab, TPUs, PyTorch)  and runs?
Thanks for any comment