首页 最新 热门 推荐

  • 首页
  • 最新
  • 热门
  • 推荐

【部署】ubuntu部署olmOCR

  • 25-04-25 04:00
  • 3991
  • 5541
blog.csdn.net

目录

  • 一、安装依赖
  • 二、安装conda新环境和sglang
  • 三、PDF解析
    • 1. 运行
    • 2. 原始 PDF 并排查看结果
    • 3. 更换模型
  • 四、可能出现的问题
    • 1.note: This error originates from a subprocess, and is likely not a problem with pip.
    • 2.转换单个PDF命令运行时
    • 3.ImportError: libnccl.so.2: cannot open shared object file: No such file or directory
    • 4.已经执行了正常命令,但是需要下载sglang

项目地址:
https://github.com/allenai/olmocr

一、安装依赖


sudo apt-get update
sudo apt-get install poppler-utils ttf-mscorefonts-installer msttcorefonts fonts-crosextra-caladea fonts-crosextra-carlito gsfonts lcdf-typetools

  • 1
  • 2
  • 3

二、安装conda新环境和sglang


conda create -n olmocr python=3.11
conda activate olmocr

git clone https://github.com/allenai/olmocr.git
cd olmocr
# If running on CPU, run this:
pip install -e .
# 如果没有权限,可以安装到用户目录下
pip install --user -e .

# If running on GPU, run this instead:
pip install -e .[gpu] --find-links https://flashinfer.ai/whl/cu124/torch2.4/flashinfer/
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12

GPU下使用还需要安装sglang

pip install --upgrade pip
pip install uv
uv pip install "sglang[all]>=0.4.4" --find-links https://flashinfer.ai/whl/cu124/torch2.5/flashinfer-python
  • 1
  • 2
  • 3

如果网络问题装不上也可以手动安装wheel,下载好wheel文件之后,sftp传到服务器上,然后直接下载

# 在这里下载
https://github.com/flashinfer-ai/flashinfer/releases/download/v0.2.3/flashinfer_python-0.2.3+cu124torch2.5-cp38-abi3-linux_x86_64.whl
  • 1
  • 2

特别慢这边
在这里插入图片描述
安装好之后再安装sglang

uv pip install "sglang[all]>=0.4.4"
  • 1

在这里插入图片描述
启动运行命令之后会安装模型权重文件,十几个G,还是很大的。

在这里插入图片描述

三、PDF解析

1. 运行

python -m olmocr.pipeline ./localworkspace --pdfs tests/gnarly_pdfs/horribleocr.pdf
  • 1

测试了两个pdf文件
在这里插入图片描述
在这里插入图片描述

在这里插入图片描述
输出的是.jsonl,所以看起来并没有换行,看的不清楚
在这里插入图片描述

转换多个PDF的命令

python -m olmocr.pipeline ./localworkspace --pdfs tests/gnarly_pdfs/*.pdf
  • 1

2. 原始 PDF 并排查看结果

python -m olmocr.viewer.dolmaviewer localworkspace/results/output_*.jsonl
  • 1

在这里插入图片描述

在这里插入图片描述

3. 更换模型

命令中可以增加模型的更换

  --model MODEL         List of paths where you can find the model to convert this pdf. You can specify several different paths here, and the script will try to use the
                        one which is fastest to access
  • 1
  • 2

四、可能出现的问题


1.note: This error originates from a subprocess, and is likely not a problem with pip.

从报错信息来看,问题出在 pip install -e . 时无法创建 olmocr.egg-info 目录,原因是 权限不足(Permission denied)。以下是解决方法:

  1. 检查当前用户权限
    确保你对项目目录(/data/huyuqiang/llf/ocr/olmocr)有写权限。可以通过以下命令检查:
ls -ld /data/huyuqiang/llf/ocr/olmocr
  • 1

如果权限不足,可以尝试修改目录权限:

sudo chown -R $USER:$USER /data/huyuqiang/llf/ocr/olmocr
sudo chmod -R u+rw /data/huyuqiang/llf/ocr/olmocr
  • 1
  • 2
  1. 使用虚拟环境
    如果你没有使用虚拟环境,建议创建一个虚拟环境并激活它,这样可以避免权限问题:
python -m venv venv  # 创建虚拟环境
source venv/bin/activate  # 激活虚拟环境
pip install -e .  # 在虚拟环境中安装
  • 1
  • 2
  • 3
  1. 指定 --user 选项
    如果你没有管理员权限,可以尝试使用 --user 选项安装到用户目录:
pip install --user -e .
  • 1

2.转换单个PDF命令运行时

File “/data/huyuqiang/llf/ocr/olmocr/olmocr/pipeline.py”, line 26, in
import torch
File “/data/huyuqiang/anaconda3/envs/olmocr/lib/python3.11/site-packages/torch/init.py”, line 367, in
from torch._C import * # noqa: F403
^^^^^^^^^^^^^^^^^^^^^^
ImportError: libcupti.so.12: cannot open shared object file: No such file or directory

在这里插入图片描述
这个错误表示系统找不到 CUDA Profiling Tools Interface (CUPTI) 库文件 libcupti.so.12。这是 CUDA 工具包的一部分,通常在安装 CUDA toolkit 时应该包含这个库。

sudo apt-get install cuda-toolkit-12-0
# 编辑 ~/.bashrc 文件
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc
sudo ln -s /usr/local/cuda/extras/CUPTI/lib64/libcupti.so.12 /usr/local/lib/libcupti.so.12
  • 1
  • 2
  • 3
  • 4
  • 5

3.ImportError: libnccl.so.2: cannot open shared object file: No such file or directory

这个错误表明在运行代码时,PyTorch 无法找到 libnccl.so.2 这个共享库文件。libnccl.so.2 是 NVIDIA 的 NCCL(NVIDIA Collective Communications Library)库的一部分,通常用于多 GPU 通信。以下是解决这个问题的步骤:

  1. 检查是否安装了 NCCL
    NCCL 是 NVIDIA 的库,通常与 CUDA 一起安装。你可以通过以下命令检查是否安装了 NCCL:
ls /usr/lib/x86_64-linux-gnu/libnccl*

  • 1
  • 2
  1. 安装 NCCL
    如果 NCCL 未安装,可以通过以下步骤安装:

方法 1:通过 NVIDIA 官方安装
访问 NCCL 下载页面。

根据你的系统(Ubuntu/CentOS 等)和 CUDA 版本下载对应的 NCCL 包。

安装下载的包。例如,对于 Ubuntu:

在这里插入图片描述

conda install -c nvidia nccl
  • 1

4.已经执行了正常命令,但是需要下载sglang

在这里插入图片描述
https://docs.sglang.ai/start/install.html

pip install --upgrade pip
pip install uv
uv pip install "sglang[all]>=0.4.4" --find-links https://flashinfer.ai/whl/cu124/torch2.5/flashinfer-python
  • 1
  • 2
  • 3
注:本文转载自blog.csdn.net的海绵波波107的文章"https://blog.csdn.net/qq_43920838/article/details/146199229"。版权归原作者所有,此博客不拥有其著作权,亦不承担相应法律责任。如有侵权,请联系我们删除。
复制链接
复制链接
相关推荐
发表评论
登录后才能发表评论和回复 注册

/ 登录

评论记录:

未查询到任何数据!
回复评论:

分类栏目

后端 (14832) 前端 (14280) 移动开发 (3760) 编程语言 (3851) Java (3904) Python (3298) 人工智能 (10119) AIGC (2810) 大数据 (3499) 数据库 (3945) 数据结构与算法 (3757) 音视频 (2669) 云原生 (3145) 云平台 (2965) 前沿技术 (2993) 开源 (2160) 小程序 (2860) 运维 (2533) 服务器 (2698) 操作系统 (2325) 硬件开发 (2492) 嵌入式 (2955) 微软技术 (2769) 软件工程 (2056) 测试 (2865) 网络空间安全 (2948) 网络与通信 (2797) 用户体验设计 (2592) 学习和成长 (2593) 搜索 (2744) 开发工具 (7108) 游戏 (2829) HarmonyOS (2935) 区块链 (2782) 数学 (3112) 3C硬件 (2759) 资讯 (2909) Android (4709) iOS (1850) 代码人生 (3043) 阅读 (2841)

热门文章

120
运维
关于我们 隐私政策 免责声明 联系我们
Copyright © 2020-2024 蚁人论坛 (iYenn.com) All Rights Reserved.
Scroll to Top