docker搭建飞桨环境(下)

安装gpu docker

  1. sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
  2. sudo nvidia-ctk runtime configure --runtime=docker

下载镜像

  1. docker pull nvidia/cuda:10.0-cudnn7-devel 查看自己的版本
  2. 创建容器:1. nvidia-docker run -it --runtime=nvidia -v /宿主机绝对路径目录:/容器内目录 --name 容器名 镜像名 /bin/bash

安装python

#!/bin/bash
set -e
apt-get update
#安装sqlite3
apt-get install libsqlite3-dev
apt-get -y install gcc
#解决ssl No module named _ssl
apt-get install libssl-dev -y
apt-get install make
apt-get -y install zlib*
#解决后续No module named '_ctypes'错误
apt install libffi-dev -y
#安装wget
apt-get -y install wget
#安装python
wget https://www.python.org/ftp/python/3.8.9/Python-3.8.9.tgz
tar -xvzf Python-3.8.9.tgz
cd Python-3.8.9
./configure --with-ssl
make && make install
ln -sf /usr/local/bin/python3 /usr/bin/python
#解决opencv 报错
apt-get -y install libgl1-mesa-glx
apt-get install -y libglib2.0-dev 
#安装pip
apt-get install python-setuptools
apt-get install python-pip

安装飞桨cpu版本

shapely
scikit-image
imgaug
pyclipper
lmdb
tqdm
numpy
visualdl
rapidfuzz
opencv-python==4.2.0.32
opencv-contrib-python==4.2.0.32
cython
lxml
premailer
openpyxl
attrdict
Polygon3
lanms-neo==1.0.2
PyMuPDF<1.21.0

requirements.txt

  1. pip3 install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
  2. pip3 install paddleocr
  3. pip3 install yacs gnureadline paddlenlp==2.2.1
  4. pip3 install xlsxwriter
  5. pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple paddlepaddle (安装时protobuf抱错版本问题不用理会-测试可以使用就行)
from paddleocr import PaddleOCR, draw_ocr
ocr  = PaddleOCR(use_angle_cls=True, lang="ch") # 会下载模型
img_path = "" # 测试图片路径
result  = ocr.ocr(img_path, cls=True)
for i in result:
    print(i)

安装飞桨Gpu版本和FastDeploy包

  1. 降级numpy pip3 install numpy==1.22
  2. 建议卸载 paddlepaddle :pip3 uninstall paddlepaddle
  3. 安装:pip3 install --upgrade paddlepaddle-gpu==2.2.0 -i https://mirror.baidu.com/pypi/simple
  4. 更新 gcc :apt-get install g++
  5. https://cmake.org/download/
  6. 下载cmake
tar  -zxvf cmake-3.26.1.tar.gz
cd cmake-3.26.1
./bootstrap # 检查环境
make
make  install
# 如果出现抱错可能需要补充的环境
apt-get install build-essential
cmake --version # 检查
  1. cmake部分参考链接:(122条消息) ubuntu安装cmake-CSDN博客
  2. 下载 TensorRT(根据自己服务器的需求来)https://developer.nvidia.com/tensorrt
  3. 下载FastDeploy文件
git clone https://github.com/PaddlePaddle/FastDeploy.git
cd FastDeploy/python
export ENABLE_ORT_BACKEND=ON
export ENABLE_PADDLE_BACKEND=ON
export ENABLE_OPENVINO_BACKEND=ON
export ENABLE_VISION=ON
export ENABLE_TEXT=ON
export ENABLE_TRT_BACKEND=ON
export WITH_GPU=ON
export TRT_DIRECTORY=/Paddle/TensorRT-8.4.1.5 #根据自己的安装报位置
export CUDA_DIRECTORY=/usr/local/cuda # find一下 虽然基本是这个位置 

python setup.py build
python setup.py bdist_wheel

pip3 install FastDeploy/python/dist/*.whl
posted @   没头发就没烦恼  阅读(240)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 25岁的心里话
· 闲置电脑爆改个人服务器(超详细) #公网映射 #Vmware虚拟网络编辑器
· 零经验选手,Compose 一天开发一款小游戏!
· 因为Apifox不支持离线,我果断选择了Apipost!
· 通过 API 将Deepseek响应流式内容输出到前端
点击右上角即可分享
微信分享提示