From 79c8b31a1e46c9d45d6629a527e81242da541089 Mon Sep 17 00:00:00 2001
From: Anirudh M <4577101+asthinasthi@users.noreply.github.com>
Date: Sun, 21 Sep 2025 04:16:26 -0700
Subject: [PATCH] Add note about cuda version while installing torch

---
 README.md | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/README.md b/README.md
index 3f6cfc2ed618..921d5449464b 100644
--- a/README.md
+++ b/README.md
@@ -229,6 +229,8 @@ This is the command to install the Pytorch xpu nightly which might have some per
 
 Nvidia users should install stable pytorch using this command:
 
+> **NOTE**: The --index-url depends on your CUDA version. 
+
 ```pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu129```
 
 This is the command to install pytorch nightly instead which might have performance improvements.