在谷歌的竞争,我已经尝试了所有三个运行时: CPU,GPU,TPU。都犯了同样的错误。# NB: Only run in TPU environment
!pip install cloud-tpu-client==0.10 https://storage.googleapis.com/tpu-pytorch/wheels/torch_xla-1.8-cp37pip -q install pytorch-lightning==1
torch_xla.core.xla_model as xm
ModuleNotFoundError: No module named 'torch_xla' 我用谷歌搜索了一下,发现"torch_xla“是用来训练TPU上的pytorch模型的东西。但我想在cpu上本地运行它(当然是为了推断),当pytorch试图加载tpu绑定张量时得到这个错误。我怎么才能修复它?