PaddlePaddle >= 2.1.2
Python 3.7
anacoda新建虚拟环境
conda create --name paddle_env python=3.8 --channel https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
进入虚拟环境
conda activate paddle_env
python -m pip install paddlepaddle-gpu==2.5.2 -i https://mirror.baidu.com/pypi/simple
pip install paddleocr>=2.0.1
pip install paddlenlp==2.5.2
安装CUDA 及对应版本的cuDNN,paddle 对驱动版本限制很死
paddle-gpu 版本要多尝试,建议老一点的版本(CPU 版本的 paddle 不影响)
下好CUDA安装包之后,会让你选个安装包解压位置,一般按默认的就好,除非C盘空间紧张,解压的内容在重启后是会自动清除的,一般问题不大
cuDNN解压后,把压缩包中cuda文件夹下的所有内容与之前 CUDA的安装路径合并
再添加PATH环境变量
----验证测试
(paddle_env) C:\Users\w6168\PaddleOCR>python
>>> import paddle
>>> paddle.utils.run_check()
参照
OCR发票关键信息抽取 - 飞桨AI Studio星河社区 (baidu.com)
修改配置,开始训练
python tools/train.py -c configs\kie\vi_layoutxlm\ser_vi_layoutxlm_xfund_zh_udml.yml