diff --git a/README.md b/README.md
index f341619..187741f 100644
--- a/README.md
+++ b/README.md
@@ -38,7 +38,7 @@ cd vggt
 pip install -r requirements.txt
 ```
 
-Alternatively, you can install VGGT as a package (<a href="docs/installation.md">Click here</a> for details).
+Alternatively, you can install VGGT as a package (<a href="docs/package.md">click here</a> for details).
 
 
 Now, try the model with just a few lines of code:
diff --git a/docs/package.md b/docs/package.md
index ecb3a85..356df89 100644
--- a/docs/package.md
+++ b/docs/package.md
@@ -7,6 +7,7 @@ This document explains how to install VGGT as a package using different package
 Before installing VGGT as a package, you need to install PyTorch and torchvision. We don't list these as dependencies to avoid CUDA version mismatches. Install them first, with an example as:
 
 ```bash
+# install pytorch 2.3.1 with cuda 12.1
 pip install torch==2.3.1 torchvision==0.18.1 --index-url https://download.pytorch.org/whl/cu121
 ```