浏览代码

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 年之前
父节点
当前提交
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.  
 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
 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 11.8版本的安装命令,更多信息请参考 https://pytorch.org/get-started/locally/  
+> ❗️务必在命令中指定以下版本
+> ```bash
+> torch==2.3.1 torchvision==0.18.1 
+> ```
+> 这是我们支持的最高版本,如果不指定版本会自动安装更高版本导致程序无法运行
+ 
 ```bash
 pip install --force-reinstall torch==2.3.1 torchvision==0.18.1 --index-url https://download.pytorch.org/whl/cu118
 ```