From b24ee2131f59c3e41f128923ad65c8821a9f603c Mon Sep 17 00:00:00 2001
From: Chenhao Jiang <jiangchenhao2@outlook.com>
Date: Sat, 19 Aug 2023 22:35:07 -0400
Subject: [PATCH] Specify cupy versions according to CUDA versions

---
 docs/install.rst | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/docs/install.rst b/docs/install.rst
index ad35851f5..5ca1374c7 100644
--- a/docs/install.rst
+++ b/docs/install.rst
@@ -20,7 +20,10 @@ Regardless of installing from wheels or from source, there are a few prerequisit
     # Update pip
     pip3 install --upgrade pip
 
-    # Install cupy
+    # Install cupy for CUDA 11.1
+    pip3 install cupy-cuda111
+
+    # Install cupy for CUDA 11.2~11.x
     pip3 install cupy-cuda11x
 
   Then, check whether your system already has NCCL installed.