docker搭建飞桨环境(下)
安装gpu docker
- sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
- sudo nvidia-ctk runtime configure --runtime=docker
下载镜像
- docker pull nvidia/cuda:10.0-cudnn7-devel 查看自己的版本
- 创建容器: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
- pip3 install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
- pip3 install paddleocr
- pip3 install yacs gnureadline paddlenlp==2.2.1
- pip3 install xlsxwriter
- 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包
- 降级numpy pip3 install numpy==1.22
- 建议卸载 paddlepaddle :pip3 uninstall paddlepaddle
- 安装:pip3 install --upgrade paddlepaddle-gpu==2.2.0 -i https://mirror.baidu.com/pypi/simple
- 更新 gcc :apt-get install g++
- https://cmake.org/download/
- 下载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 # 检查
- cmake部分参考链接:(122条消息) ubuntu安装cmake-CSDN博客
- 下载 TensorRT(根据自己服务器的需求来)https://developer.nvidia.com/tensorrt
- 下载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
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 25岁的心里话
· 闲置电脑爆改个人服务器(超详细) #公网映射 #Vmware虚拟网络编辑器
· 零经验选手,Compose 一天开发一款小游戏!
· 因为Apifox不支持离线,我果断选择了Apipost!
· 通过 API 将Deepseek响应流式内容输出到前端