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
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
```
......
......@@ -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
```
......
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