Updated colab tutorials to PyTorch 1.11 (#3428)
* updated colab tutorials to PyTorch1.11
* migrating tutorials to PyTorch 1.11
* uncommented cached-dataset pip
* Updated all the tutorials to PyTorch 1.11 and fixed formatting issue.
* Moved from update_env_setup_script branch to master for torch xla wheel
