Commit fd60393d authored by myhloli's avatar myhloli

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.
parent 6dbb4197
...@@ -151,6 +151,11 @@ If you have an available Nvidia GPU or are using a Mac with Apple Silicon, you c ...@@ -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
``` ```
......
...@@ -140,6 +140,12 @@ cp magic-pdf.template.json ~/magic-pdf.json ...@@ -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
``` ```
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment