#TPU #PyTorch #Jax
text
#TPU #PyTorch #Jax
new_line
new_line
When You're trying to use PyTorch or Jax on TPU,
text
When You're trying to use PyTorch or Jax on TPU,
new_line
new_line
for v2/v3/v4:
text
for v2/v3/v4:
new_line
  use tpu-ubuntu2204-base
text
  use tpu-ubuntu2204-base
new_line
new_line
for v5p:
text
for v5p:
new_line
  use v2-alpha-tpuv5
text
  use v2-alpha-tpuv5
new_line
new_line
for v5e:
text
for v5e:
new_line
  use v2-alpha-tpuv5-lite
text
  use v2-alpha-tpuv5-lite
new_line
new_line
You must use these base images for the system to 'boot'.
text
You must use these base images for the system to 'boot'.
new_line
new_line
Previously used tpu-vm-v4-pt-1.13 images might seem to start the VM, but SSH connections do not work.
text
Previously used tpu-vm-v4-pt-1.13 images might seem to start the VM, but SSH connections do not work.
new_line
new_line
I thought it was a firewall issue and spent a lot of time on it before realizing it was a problem with the boot image 🥲
text
I thought it was a firewall issue and spent a lot of time on it before realizing it was a problem with the boot image 🥲
new_line
new_line
https://cloud.google.com/tpu/docs/runtimes#pytorch_and_jax
https://cloud.google.com/tpu/docs/runtimes#pytorch_and_jax
link
#TPU #PyTorch #Jax

When You're trying to use PyTorch or Jax on TPU,

for v2/v3/v4:
  use tpu-ubuntu2204-base

for v5p:
  use v2-alpha-tpuv5

for v5e:
  use v2-alpha-tpuv5-lite

You must use these base images for the system to 'boot'.

Previously used tpu-vm-v4-pt-1.13 images might seem to start the VM, but SSH connections do not work.

I thought it was a firewall issue and spent a lot of time on it before realizing it was a problem with the boot image 🥲

https://cloud.google.com/tpu/docs/runtimes#pytorch_and_jax
👍
deepkyu maywell devngho byoussef Taekyoon DevWorld