Lesson 8 00_load_data.ipynb swift version crash

I was trying to run lesson 8 00_load_data.ipynb of part 2 2019 swift version in Colab but it crash when I get to import TensorFlow. May be it’s missing TensorFlow package? Has anyone run into this issue before?

I run this nb in colab and it work good.
Did you change runtime type to GPU?