@Redwindsurya web demo 的 docker 版(很强的OCR模型) 中发帖

surya web demo 的 docker 版
之前在 L 站看到了有人分享这个模型,但是我自己作为外行搞了一晚上也没解决在 Windows 里的环境问题,怎么调都是在 CPU 上跑,今天试着在 GPT 的帮助下写了个 dockerfile,终于能在显卡上跑了,速度快了很多。不多墨迹,直接上文件:
# 使用 pytorch/pytorch:2.4.1-cuda12.4-cudnn9-runtime 作为基础镜像
FROM pytorch/pytorch:2.4.1-cuda12.4-cudnn9-runtime

# 设置工作目录
WORKDIR /app

# 安装必要的依赖库
RUN apt-get update && apt-get install -y \
python3-pip \
libgl1-mesa-glx \
libsm6 \
lib...