From ede7d361a4f085db7d023c1cf5c121b011d16ef5 Mon Sep 17 00:00:00 2001 From: myhloli Date: Mon, 9 Dec 2024 18:11:24 +0800 Subject: [PATCH] docs(windows): update CUDA installation guide - Remove specific version requirements for torch and torchvision - Simplify installation command in both English and Chinese guides - Delete important note about version compatibility --- docs/README_Windows_CUDA_Acceleration_en_US.md | 10 +--------- docs/README_Windows_CUDA_Acceleration_zh_CN.md | 11 +---------- 2 files changed, 2 insertions(+), 19 deletions(-) diff --git a/docs/README_Windows_CUDA_Acceleration_en_US.md b/docs/README_Windows_CUDA_Acceleration_en_US.md index 73a2de5c..e8252145 100644 --- a/docs/README_Windows_CUDA_Acceleration_en_US.md +++ b/docs/README_Windows_CUDA_Acceleration_en_US.md @@ -65,16 +65,8 @@ If your graphics card has at least 8GB of VRAM, follow these steps to test CUDA- 1. **Overwrite the installation of torch and torchvision** supporting CUDA. ``` - 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 torchvision --index-url https://download.pytorch.org/whl/cu118 ``` - > [!IMPORTANT] - > Ensure the following versions are specified in the command: - > - > ``` - > torch==2.3.1 torchvision==0.18.1 - > ``` - > - > These are the highest versions we support. Installing higher versions without specifying them will cause the program to fail. 2. **Modify the value of `"device-mode"`** in the `magic-pdf.json` configuration file located in your user directory. diff --git a/docs/README_Windows_CUDA_Acceleration_zh_CN.md b/docs/README_Windows_CUDA_Acceleration_zh_CN.md index d0454f7d..0b156994 100644 --- a/docs/README_Windows_CUDA_Acceleration_zh_CN.md +++ b/docs/README_Windows_CUDA_Acceleration_zh_CN.md @@ -66,18 +66,9 @@ pip install -U magic-pdf[full] --extra-index-url https://wheels.myhloli.com -i h **1.覆盖安装支持cuda的torch和torchvision** ```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 torchvision --index-url https://download.pytorch.org/whl/cu118 ``` -> [!IMPORTANT] -> 务必在命令中指定以下版本 -> -> ```bash -> torch==2.3.1 torchvision==0.18.1 -> ``` -> -> 这是我们支持的最高版本，如果不指定版本会自动安装更高版本导致程序无法运行 - **2.修改【用户目录】中配置文件magic-pdf.json中"device-mode"的值** ```json