From 7d9a69e8d8d724378da3520cf694e61d0db86896 Mon Sep 17 00:00:00 2001
From: guhaopeng <1802702972@qq.com>
Date: Fri, 12 Sep 2025 17:07:23 +0800
Subject: [PATCH] docs: Update README.md

An additional line of comments has been added to make the code clearer.
---
 README.md | 1 +
 1 file changed, 1 insertion(+)

diff --git a/README.md b/README.md
index 73e5e1d..9dad6b3 100755
--- a/README.md
+++ b/README.md
@@ -53,6 +53,7 @@ Install dependencies:
 ```bash
 # pytorch (select correct CUDA version)
 pip install torch torchvision --index-url https://download.pytorch.org/whl/{your-cuda-version}
+#if your cuda version is 11.8,this code shoule be ：pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118 
 
 # other dependencies
 pip install -r requirements.txt