diff --git a/README.md b/README.md index e31f234a6..f17fd1d79 100644 --- a/README.md +++ b/README.md @@ -1,7 +1,9 @@ #### TPU/XLA Devices Users with TPU/XLA devices can install the PyTorch XLA stable build with the following command: -```pip install torch~=2.5.0 torch_xla[tpu]~=2.5.0 -f https://storage.googleapis.com/libtpu-releases/index.html``` +``` +pip install torch~=2.5.0 torch_xla[tpu]~=2.5.0 -f https://storage.googleapis.com/libtpu-releases/index.html +``` This is the command to install the nightly 2.6.0 which might have some performance improvements: @@ -13,4 +15,6 @@ pip install 'torch_xla[tpu] @ https://storage.googleapis.com/pytorch-xla-release To get memory info for TPU devices, install the [tpu-info](https://github.com/AI-Hypercomputer/cloud-accelerator-diagnostics/tree/main/tpu_info) package with the following command: -```pip install tpu-info``` +``` +pip install tpu-info +```