README_Windows_CUDA_Acceleration_en_US.md 2.52 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
# Windows 10/11

### 1. Install CUDA and cuDNN

Required versions: CUDA 11.8 + cuDNN 8.7.0

- CUDA 11.8: https://developer.nvidia.com/cuda-11-8-0-download-archive
- cuDNN v8.7.0 (November 28th, 2022), for CUDA 11.x: https://developer.nvidia.com/rdp/cudnn-archive

### 2. Install Anaconda

If Anaconda is already installed, you can skip this step.

Download link: https://repo.anaconda.com/archive/Anaconda3-2024.06-1-Windows-x86_64.exe

### 3. Create an Environment Using Conda

Python version must be 3.10.

```
conda create -n MinerU python=3.10
conda activate MinerU
```

### 4. Install Applications

```
pip install -U magic-pdf[full] --extra-index-url https://wheels.myhloli.com
```

31
32
> [!IMPORTANT]
> After installation, verify the version of `magic-pdf`:
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
>
> ```bash
> magic-pdf --version
> ```
>
> If the version number is less than 0.7.0, please report it in the issues section.

### 5. Download Models

Refer to detailed instructions on [how to download model files](how_to_download_models_en.md).

### 6. Understand the Location of the Configuration File

After completing the [5. Download Models](#5-download-models) step, the script will automatically generate a `magic-pdf.json` file in the user directory and configure the default model path.
You can find the `magic-pdf.json` file in your 【user directory】 .

49
> [!TIP]
50
51
52
53
54
55
56
57
> The user directory for Windows is "C:/Users/username".

### 7. First Run

Download a sample file from the repository and test it.

```powershell
  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
62
63
64
65
66
67
```

### 8. Test CUDA Acceleration

If your graphics card has at least 8GB of VRAM, follow these steps to test CUDA-accelerated parsing performance.

1. **Overwrite the installation of torch and torchvision** supporting CUDA.

   ```
68
   pip install --force-reinstall torch==2.3.1 torchvision==0.18.1 --index-url https://download.pytorch.org/whl/cu118
69
70
71
72
73
74
75
76
77
78
   ```

2. **Modify the value of `"device-mode"`** in the `magic-pdf.json` configuration file located in your user directory.

   ```json
   {
     "device-mode": "cuda"
   }
   ```

79

80
81
82
3. **Run the following command to test CUDA acceleration**:

   ```
83
   magic-pdf -p small_ocr.pdf -o ./output
84
85
86
87
88
89
90
91
92
93
   ```

### 9. Enable CUDA Acceleration for OCR

1. **Download paddlepaddle-gpu**, which will automatically enable OCR acceleration upon installation.
   ```
   pip install paddlepaddle-gpu==2.6.1
   ```
2. **Run the following command to test OCR acceleration**:
   ```
94
   magic-pdf -p small_ocr.pdf -o ./output
95
   ```