docker-搭建飞桨环境（下）

不想秃头的新人

已于 2023-03-30 17:41:02 修改

阅读量190

点赞数

文章标签： docker paddlepaddle python

于 2023-03-30 15:09:51 首次发布

本文链接：https://blog.csdn.net/weixin_46187083/article/details/129858219

版权

安装飞桨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