TPU setup in docs results in import failure for flash.Task
I am following TPU setup instructions on a clean Colab notebook. When I do so and attempt to import flash.Task I get the following error:
ImportError: /usr/local/lib/python3.7/dist-packages/_XLAC.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZN2at13_foreach_add_EN3c108ArrayRefINS_6TensorEEENS1_INS0_6ScalarEEE
This is being run on Colab Pro with TPU enabled.