From 6596447f16acd26f4e78b566d171b7355624aa60 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= Date: Fri, 12 Mar 2021 03:04:47 +0100 Subject: [PATCH] update xla version (#6464) --- notebooks/06-mnist-tpu-training.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/notebooks/06-mnist-tpu-training.ipynb b/notebooks/06-mnist-tpu-training.ipynb index 5d55388b2efcf..a0dfdceece9b1 100644 --- a/notebooks/06-mnist-tpu-training.ipynb +++ b/notebooks/06-mnist-tpu-training.ipynb @@ -80,7 +80,7 @@ "id": "AYGWh10lRaF1" }, "source": [ - "! pip install cloud-tpu-client==0.10 https://storage.googleapis.com/tpu-pytorch/wheels/torch_xla-1.7-cp37-cp37m-linux_x86_64.whl" + "! pip install cloud-tpu-client==0.10 https://storage.googleapis.com/tpu-pytorch/wheels/torch_xla-1.8-cp37-cp37m-linux_x86_64.whl" ], "execution_count": null, "outputs": []