dekun/secondary-school-grade-archive
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

OCR 修复：无 cuDNN 用 CPU、禁用 ir_optim 避免 SIGILL。

Browse Source
This commit is contained in:
dekun
2026-06-28 15:18:42 +08:00
parent 0d4861fa62
commit 035b65dcc8
3 changed files with 39 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
deploy/ocr-worker/install.sh
+5 -5
View File
@@ -25,23 +25,23 @@ source "${VENV}/bin/activate"
pip install -U pip wheel -i "${PIP_MIRROR}"
install_paddle() {
if command -v nvidia-smi >/dev/null 2>&1; then
local use_gpu="${OCR_USE_GPU:-false}"
if [[ "${use_gpu}" == "true" ]] && command -v nvidia-smi >/dev/null 2>&1 && ldconfig -p 2>/dev/null | grep -q libcudnn; then
local cuda_major
cuda_major="$(nvidia-smi 2>/dev/null | sed -n 's/.*CUDA Version: \([0-9]*\)\.[0-9]*/\1/p' | head -1)"
cuda_major="${cuda_major:-11}"
echo "==> 检测到 NVIDIA GPU，CUDA 主版本: ${cuda_major}"
echo "==> 安装 paddlepaddle-gpu (CUDA ${cuda_major}.x)…"
if [[ "${cuda_major}" -ge 12 ]]; then
echo "==> 安装 paddlepaddle-gpu (CUDA 12.x)…"
pip install paddlepaddle-gpu==2.6.2 -i https://www.paddlepaddle.org.cn/packages/stable/cu123/ \
|| pip install paddlepaddle-gpu==2.6.2 -i https://www.paddlepaddle.org.cn/packages/stable/cu118/
else
echo "==> 安装 paddlepaddle-gpu (CUDA 11.x)…"
pip install paddlepaddle-gpu==2.6.2 -i https://www.paddlepaddle.org.cn/packages/stable/cu118/
fi
else
echo "==> 未检测到 GPU，安装 CPU 版 paddlepaddle…"
echo "==> 安装 paddlepaddle CPU（无 cuDNN 或 OCR_USE_GPU=false）…"
pip install paddlepaddle==2.6.2 -i "${PIP_MIRROR}"
fi
pip install 'protobuf>=3.20,<4' -q
}
install_paddle