Import error on using torch_XLA

IamSparky: <p>As I am working on TPU kernal,</p>
<p>I have executed these lines<br>
<div class="lightbox-wrapper"><a class="lightbox" href="https://discuss.pytorch.org/uploads/default/original/3X/9/1/91cfa38021d6a0f3d2475d919c77978be98f598d.png" data-download-href="https://discuss.pytorch.org/uploads/default/91cfa38021d6a0f3d2475d919c77978be98f598d" title="image"><img src="https://discuss.pytorch.org/uploads/default/optimized/3X/9/1/91cfa38021d6a0f3d2475d919c77978be98f598d_2_690x384.png" alt="image" data-base62-sha1="kNU60X7a49Ft2sWfzpL06e7bbat" width="690" height="384" srcset="https://discuss.pytorch.org/uploads/default/optimized/3X/9/1/91cfa38021d6a0f3d2475d919c77978be98f598d_2_690x384.png, https://discuss.pytorch.org/uploads/default/optimized/3X/9/1/91cfa38021d6a0f3d2475d919c77978be98f598d_2_1035x576.png 1.5x, https://discuss.pytorch.org/uploads/default/original/3X/9/1/91cfa38021d6a0f3d2475d919c77978be98f598d.png 2x" data-dominant-color="4B4D52"><div class="meta"><svg class="fa d-icon d-icon-far-image svg-icon" aria-hidden="true"><use href="#far-image"></use></svg><span class="filename">image</span><span class="informations">1228×684 95.9 KB</span><svg class="fa d-icon d-icon-discourse-expand svg-icon" aria-hidden="true"><use href="#discourse-expand"></use></svg></div></a></div></p>
<p>and then got this error on running these imports<br>
<div class="lightbox-wrapper"><a class="lightbox" href="https://discuss.pytorch.org/uploads/default/original/3X/d/0/d06ab817ed0f8606f99a0412bd32598a2ded33d9.jpeg" data-download-href="https://discuss.pytorch.org/uploads/default/d06ab817ed0f8606f99a0412bd32598a2ded33d9" title="wqwqw"><img src="https://discuss.pytorch.org/uploads/default/optimized/3X/d/0/d06ab817ed0f8606f99a0412bd32598a2ded33d9_2_690x259.jpeg" alt="wqwqw" data-base62-sha1="tJJXqu5V1h2W6S2Kk9A66Y1prcZ" width="690" height="259" srcset="https://discuss.pytorch.org/uploads/default/optimized/3X/d/0/d06ab817ed0f8606f99a0412bd32598a2ded33d9_2_690x259.jpeg, https://discuss.pytorch.org/uploads/default/optimized/3X/d/0/d06ab817ed0f8606f99a0412bd32598a2ded33d9_2_1035x388.jpeg 1.5x, https://discuss.pytorch.org/uploads/default/optimized/3X/d/0/d06ab817ed0f8606f99a0412bd32598a2ded33d9_2_1380x518.jpeg 2x" data-dominant-color="474A51"><div class="meta"><svg class="fa d-icon d-icon-far-image svg-icon" aria-hidden="true"><use href="#far-image"></use></svg><span class="filename">wqwqw</span><span class="informations">1442×542 71.1 KB</span><svg class="fa d-icon d-icon-discourse-expand svg-icon" aria-hidden="true"><use href="#discourse-expand"></use></svg></div></a></div></p>
<p><a class="mention-group" href="/groups/admins">@admins</a> <a class="mention" href="/u/smth">@smth</a> <a class="mention" href="/u/ptrblck">@ptrblck</a> please help.</p>

ptrblck: <p>Please don’t tag certain people, as this might discourage others to answer and might also tag users, who are not experts in the topic.</p>
<p>That being said, I’ll move your topic to the XLA category to add visibility.</p>

IamSparky: <p>Thanks ptrblck … can please suggest some tips and tricks to add visibility to my topics while posting? I would very greatful as I am new to pytorch forums.</p>

ptrblck: <p>I’ve already moved the topic to the right category, so let’s wait until <a class="mention" href="/u/ailzhang">@ailzhang</a> and other XLA expert have some time to look into the topic. As it’s weekend, you might need to wait a bit. <img src="https://discuss.pytorch.org/images/emoji/apple/wink.png?v=9" title=":wink:" class="emoji" alt=":wink:"></p>

IamSparky: <p><a class="mention" href="/u/ailzhang">@ailzhang</a> or anyone from torch xla team if you can help please</p>

ailzhang: <p>Hi <a class="mention" href="/u/iamsparky">@IamSparky</a>, sorry for the late reply, we take all xla/tpu related questions/feature requests/bug reports as issues in pytorch/xla repo. Would you mind posting the issue in pytorch/xla github with a link to the colab notebook? I’m pretty sure this should work but it’s hard to locate without seeing the code.</p>