Эх сурвалжийг харах

fix(readme): specify supported PyTorch versions in install guide

Update the PyTorch installation guide in both English and Chinese READMEs to explicitly
recommend using torch==2.3.1 and torchvision==0.18.1 for CUDA 11.8. Emphasize the
importance of specifying these versions to avoid compatibility issues with higher,
unsupported versions.
myhloli 1 жил өмнө
parent
commit
fd60393da1
2 өөрчлөгдсөн 11 нэмэгдсэн , 0 устгасан
  1. 5 0
      README.md
  2. 6 0
      README_zh-CN.md

+ 5 - 0
README.md

@@ -151,6 +151,11 @@ If you have an available Nvidia GPU or are using a Mac with Apple Silicon, you c
 
 
 You need to install the corresponding PyTorch version according to your CUDA version.  
 You need to install the corresponding PyTorch version according to your CUDA version.  
 This example installs the CUDA 11.8 version.More information https://pytorch.org/get-started/locally/  
 This example installs the CUDA 11.8 version.More information https://pytorch.org/get-started/locally/  
+> ❗ ️Make sure to specify version
+> ```bash
+> torch==2.3.1 torchvision==0.18.1
+> ```
+>  in the command, as these are the highest versions we support. Failing to specify the versions may result in automatically installing higher versions which can cause the program to fail.
 ```bash
 ```bash
 pip install --force-reinstall torch==2.3.1 torchvision==0.18.1 --index-url https://download.pytorch.org/whl/cu118
 pip install --force-reinstall torch==2.3.1 torchvision==0.18.1 --index-url https://download.pytorch.org/whl/cu118
 ```
 ```

+ 6 - 0
README_zh-CN.md

@@ -140,6 +140,12 @@ cp magic-pdf.template.json ~/magic-pdf.json
 
 
 需要根据自己的CUDA版本安装对应的pytorch版本  
 需要根据自己的CUDA版本安装对应的pytorch版本  
 以下是对应CUDA 11.8版本的安装命令,更多信息请参考 https://pytorch.org/get-started/locally/  
 以下是对应CUDA 11.8版本的安装命令,更多信息请参考 https://pytorch.org/get-started/locally/  
+> ❗️务必在命令中指定以下版本
+> ```bash
+> torch==2.3.1 torchvision==0.18.1 
+> ```
+> 这是我们支持的最高版本,如果不指定版本会自动安装更高版本导致程序无法运行
+ 
 ```bash
 ```bash
 pip install --force-reinstall torch==2.3.1 torchvision==0.18.1 --index-url https://download.pytorch.org/whl/cu118
 pip install --force-reinstall torch==2.3.1 torchvision==0.18.1 --index-url https://download.pytorch.org/whl/cu118
 ```
 ```