From bb89040ad7f3f411c6809acb09e53ad381f370ca Mon Sep 17 00:00:00 2001
From: Xiongfei Wei <isaacwxf23@gmail.com>
Date: Mon, 18 Sep 2023 17:13:14 +0000
Subject: [PATCH] Update gpu doc.

---
 docs/gpu.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docs/gpu.md b/docs/gpu.md
index 2036ac7f742..49a4319d482 100644
--- a/docs/gpu.md
+++ b/docs/gpu.md
@@ -51,8 +51,8 @@ Thu Dec  8 06:24:29 2022
 
 ### Wheel
 ```
-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
+pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/test/cu118
+pip3 install <torch_xla_new_release_wheel. Example: https://storage.googleapis.com/tpu-pytorch/wheels/cuda/117/torch_xla-2.0-cp38-cp38-linux_x86_64.whl>
 ```
 
 ## Run a simple model