README_Windows_CUDA_Acceleration_zh_CN.md 2.31 KB
Newer Older
1
2
3
4
# Windows10/11

## 1. 安装cuda和cuDNN

5
需要安装符合torch要求的cuda版本,torch目前支持11.8/12.4/12.6
6

7
- CUDA 11.8 https://developer.nvidia.com/cuda-11-8-0-download-archive
8
9
- CUDA 12.4 https://developer.nvidia.com/cuda-12-4-0-download-archive
- CUDA 12.6 https://developer.nvidia.com/cuda-12-6-0-download-archive
10
11

## 2. 安装anaconda
12

13
14
15
如果已安装conda,可以跳过本步骤

下载链接:
16
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2024.06-1-Windows-x86_64.exe
17
18

## 3. 使用conda 创建环境
19

20
```bash
21
22
conda create -n mineru 'python<3.13' -y
conda activate mineru
23
```
24

25
## 4. 安装应用
26

27
```bash
28
pip install -U magic-pdf[full] -i https://mirrors.aliyun.com/pypi/simple
29
```
30

31
32
> [!IMPORTANT]
> 下载完成后,务必通过以下命令确认magic-pdf的版本是否正确
33
>
34
35
> ```bash
> magic-pdf --version
36
37
> ```
>
38
> 如果版本号小于 1.3.0 ,请到issue中向我们反馈
39
40

## 5. 下载模型
41

42
详细参考 [如何下载模型文件](how_to_download_models_zh_cn.md)
43

44
## 6. 了解配置文件存放的位置
45

46
47
完成[5.下载模型](#5-下载模型)步骤后,脚本会自动生成用户目录下的magic-pdf.json文件,并自动配置默认模型路径。
您可在【用户目录】下找到magic-pdf.json文件。
48

49
> [!TIP]
50
> windows用户目录为 "C:/Users/用户名"
51
52

## 7. 第一次运行
53

54
从仓库中下载样本文件,并测试
55

56
```powershell
57
 wget https://github.com/opendatalab/MinerU/raw/master/demo/small_ocr.pdf -O small_ocr.pdf
58
 magic-pdf -p small_ocr.pdf -o ./output
59
60
61
```

## 8. 测试CUDA加速
62

63
如果您的显卡显存大于等于 **6GB** ,可以进行以下流程,测试CUDA解析加速效果
64

65
**1.覆盖安装支持cuda的torch和torchvision**(请根据cuda版本选择合适的index-url,具体可参考[torch官网](https://pytorch.org/get-started/locally/))
66

67
```bash
68
pip install --force-reinstall torch==2.6.0 torchvision==0.21.1 "numpy<2.0.0" --index-url https://download.pytorch.org/whl/cu124
69
```
70

71
**2.修改【用户目录】中配置文件magic-pdf.json中"device-mode"的值**
72

73
74
75
76
77
```json
{
  "device-mode":"cuda"
}
```
78

79
**3.运行以下命令测试cuda加速效果**
80

81
```bash
82
magic-pdf -p small_ocr.pdf -o ./output
83
```
84

85
> [!TIP]
86
> CUDA加速是否生效可以根据log中输出的各个阶段的耗时来简单判断,通常情况下,cuda加速后运行速度比cpu更快。