diff --git a/docs/gpu.md b/docs/gpu.md
index 9e84860c81b..de1cf807361 100644
--- a/docs/gpu.md
+++ b/docs/gpu.md
@@ -72,8 +72,11 @@ source ~/.bashrc
 ### Wheel
 ```
 pip3 install torch==2.3.0
+# GPU whl for python 3.10 + cuda 12.1
 pip3 install https://storage.googleapis.com/pytorch-xla-releases/wheels/cuda/12.1/torch_xla-2.3.0-cp310-cp310-manylinux_2_28_x86_64.whl
 ```
+Wheels for other Python version and CUDA version can be found [here](https://github.com/pytorch/xla?tab=readme-ov-file#available-docker-images-and-wheels).
+
 
 ## Run a simple model
 In order to run below examples, you need to clone the pytorch/xla repo to access the imagenet example(We already clone it in our docker).