Fix pip install command typo in GPU doc. (#5545)

* Set XLA_PYTHON_CLIENT_MEM_FRACTION to 0.8 * Fix pip install command typo in GPU doc. * remove unwanted change
pytorch · Sep 6, 2023 · 2c8ebd1 · 2c8ebd1
1 parent 04118d5
commit 2c8ebd1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/gpu.md b/docs/gpu.md
@@ -51,7 +51,7 @@ Thu Dec  8 06:24:29 2022
 
 ### Wheel
 ```
-pip3 install torch=2.0
+pip3 install torch==2.0
 pip3 install https://storage.googleapis.com/tpu-pytorch/wheels/cuda/117/torch_xla-2.0-cp38-cp38-linux_x86_64.whl
 ```
 

diff --git a/torch_xla/__init__.py b/torch_xla/__init__.py
@@ -140,4 +140,4 @@ def _init_xla_lazy_backend():
 # TODO @wonjoo come up with a long term fix in Dynamo.
 torch._dynamo.config.automatic_dynamic_shapes = False
 
-from .stablehlo import save_as_stablehlo, save_torch_model_as_stablehlo
+from .stablehlo import save_as_stablehlo, save_torch_model_as_stablehlo