核心内容摘要
蜜芽miya：解锁你的专属网名，让数字世界闪耀个性之光
VMware虚拟机中部署DeepSeek-OCR-2的完整指南
引言在当今数字化办公环境中OCR光学字符识别技术已成为处理文档、扫描件和图片中文字信息的重要工具。
DeepSeek-OCR-2作为新一代开源OCR模型凭借其创新的视觉因果流技术在识别准确率和处理效率上都有显著提升。
本文将详细介绍如何在VMware虚拟化环境中部署DeepSeek-OCR-2包括虚拟机配置、环境搭建、模型部署和性能优化等关键步骤。
通过本教程你将学会如何配置适合DeepSeek-OCR-2运行的VMware虚拟机一步步安装和配置必要的软件环境部署DeepSeek-OCR-2模型并进行基本测试优化虚拟机资源分配以获得最佳性能
环境准备
1 硬件要求在VMware中运行DeepSeek-OCR-2需要满足以下硬件条件主机硬件CPU至少4核推荐8核或以上内存至少16GB推荐32GB存储至少50GB可用空间SSD推荐GPU如果主机有NVIDIA GPU可以启用直通功能虚拟机配置vCPU至少4个推荐8个内存至少12GB推荐16GB或以上显存如果使用GPU直通分配至少8GB显存磁盘空间至少40GB
2 软件要求VMware Workstation Pro 17 或 ESXi
0操作系统Ubuntu
2
04 LTS推荐Python
3.
1
9CUDA
1
8如果使用GPUPyTorch
2.
0Transformers
4.
46.
创建和配置虚拟机
1 新建虚拟机打开VMware Workstation点击创建新虚拟机选择自定义(高级)配置选择虚拟机硬件兼容性默认最新版本即可选择稍后安装操作系统选择Linux Ubuntu 64位设置虚拟机名称和存储位置处理器配置至少4核推荐8核内存至少12GB推荐16GB网络连接桥接模式推荐或NATI/O控制器类型默认LSI Logic磁盘类型SCSI创建新虚拟磁盘大小至少40GB选择将虚拟磁盘拆分成多个文件完成虚拟机创建
2 安装Ubuntu系统挂载Ubuntu
2
04 ISO镜像启动虚拟机并开始安装选择语言和键盘布局网络配置保持默认或按需设置磁盘分区选择使用整个磁盘新手推荐或手动分区设置用户名和密码等待安装完成并重启
3 安装VMware Tools安装VMware Tools可以改善虚拟机性能和使用体验sudo apt update sudo apt install open-vm-tools open-vm-tools-desktop sudo reboot
安装必要软件和驱动
1 更新系统和安装基础工具sudo apt update sudo apt upgrade -y sudo apt install -y git wget curl build-essential python3-pip python3-venv
2 安装NVIDIA驱动如果使用GPU直通首先检查可用的驱动版本ubuntu-drivers devices安装推荐的驱动版本sudo ubuntu-drivers autoinstall重启后验证安装nvidia-smi
3 安装CUDA Toolkit
1
8wget https://developer.download.nvidia.com/compute/cuda/
11.
0/local_installers/cuda_
11.
0_
520.
6
05_linux.run sudo sh cuda_
11.
0_
520.
6
05_linux.run安装时选择接受协议取消勾选驱动安装如果已安装确保CUDA Toolkit被选中添加环境变量到~/.bashrcecho export PATH/usr/local/cuda-
1
8/bin:$PATH ~/.bashrc echo export LD_LIBRARY_PATH/usr/local/cuda-
1
8/lib64:$LD_LIBRARY_PATH ~/.bashrc source ~/.bashrc
4 安装cuDNN从NVIDIA官网下载cuDNN for CUDA
1
8解压并安装tar -xzvf cudnn-linux-x86_64-
8.
9.
25_cuda11-archive.tar.xz sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 sudo chmod ar /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
部署DeepSeek-OCR-
2
1 创建Python虚拟环境python3 -m venv deepseek-env source deepseek-env/bin/activate
2 安装PyTorch和其他依赖pip install torch
2.
0 torchvision
0.
2
0 torchaudio
2.
0 --index-url https://download.pytorch.org/whl/cu118 pip install transformers
4.
4
3 pip install flash-attn
2.
3 --no-build-isolation pip install vllm
0.
8.
5
3 克隆DeepSeek-OCR-2仓库git clone https://github.com/deepseek-ai/DeepSeek-OCR-
git cd DeepSeek-OCR-2 pip install -r requirements.txt
运行DeepSeek-OCR-
2
1 基本使用示例创建一个简单的Python脚本test_ocr.pyfrom transformers import AutoModel, AutoTokenizer import torch import os os.environ[CUDA_VISIBLE_DEVICES] 0 model_name deepseek-ai/DeepSeek-OCR-2 tokenizer AutoTokenizer.from_pretrained(model_name, trust_remote_codeTrue) model AutoModel.from_pretrained( model_name, _attn_implementationflash_attention_2, trust_remote_codeTrue, use_safetensorsTrue ) model model.eval().cuda().to(torch.bfloat
# 文档转换到Markdown prompt image\n|grounding|Convert the document to markdown. image_file test_image.jpg # 替换为你的图片路径 output_path output # 输出目录 res model.infer( tokenizer, promptprompt, image_fileimage_file, output_pathoutput_path, base_size1024, image_size768, crop_modeTrue ) print(f识别结果已保存到: {output_path})
2 运行测试python test_ocr.py
性能优化
1 虚拟机资源配置优化CPU分配根据主机核心数合理分配vCPU启用CPU虚拟化功能VT-x/AMD-V内存优化分配足够内存至少12GB启用内存预留功能磁盘性能使用SSD存储启用磁盘预分配考虑使用独立磁盘模式GPU直通如果可用在VMware设置中启用PCI设备直通分配整个GPU给虚拟机
2 模型推理优化使用vLLM加速推理from vllm import LLM, SamplingParams llm LLM(modeldeepseek-ai/DeepSeek-OCR-
sampling_params SamplingParams(temperature
0, top_p
1.
批处理优化同时处理多个文档提高吞吐量调整batch_size参数找到最佳值量化模型使用4-bit或8-bit量化减少内存占用
8.
常见问题解决
1 CUDA内存不足解决方案减少batch_size使用更小的输入分辨率启用模型量化增加虚拟机显存分配
2 安装依赖失败解决方案确保使用正确的Python版本(
3.
12.
检查CUDA和cuDNN版本匹配尝试使用conda环境
3 模型下载慢解决方案使用国内镜像源手动下载模型文件并指定本地路径
4 虚拟机性能不佳解决方案检查主机资源使用情况关闭不必要的虚拟机服务调整虚拟机资源分配
9.
总结通过本教程我们详细介绍了在VMware虚拟化环境中部署DeepSeek-OCR-2的完整流程。
从虚拟机配置、环境搭建到模型部署和性能优化每个步骤都提供了具体的操作指南。
DeepSeek-OCR-2作为新一代OCR模型在虚拟化环境中也能发挥出色的性能特别是在处理复杂文档和表格时表现优异。
实际使用中建议根据具体应用场景调整虚拟机资源配置和模型参数以达到最佳的性能和准确率平衡。
随着项目的更新也可以关注DeepSeek-OCR-2的GitHub仓库获取最新功能和优化。