[AI]docker封装包含cuda cudnn的paddlepaddle PaddleOCR

封装,启动时需要在GPU服务器上

显卡驱动 cuda等下载需要注册账号

环境 Ubuntu 24.04 LTS
镜像 uvicorn-gunicorn-fastapi:python3.8
显卡支持版本 CUDA Version: 12.5
cuda版本 cuda_12.4.0_550.54.14_linux.run
cudnn版本
PaddlePaddle-Gpu
PaddleOCR

启动进入容器

复制代码
 docker run  --gpus all -it  tiangolo/uvicorn-gunicorn-fastapi:python3.8 --name ugf-cuda-cudnn
 docker exec it ugf-cuda-cudnn /bin/bash

安装cuda

1.下载cuda

CUDA Toolkit Archive | NVIDIA Developer
显卡支持cuda版本为12.5 cuda-toolkit降低一个版本12.4

复制代码
wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run

2.部署cuda
复制代码
sudo sh cuda_12.4.0_550.54.14_linux.run

输入accept
把这里的X按空格取消掉后再进行安装
配置环境变量

复制代码
echo 'export PATH=$PATH:/usr/local/cuda-12.4/bin' >>/etc/profile
echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-12.4/lib64' >>/etc/profile
echo 'export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-12.4' >>/etc/profile
source  /etc/profile

检查是否安装成功

如果出现下面版本号信息则安装成功

复制代码
root@ef174907636d:~# nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2024 NVIDIA Corporation
Built on Tue_Feb_27_16:19:38_PST_2024
Cuda compilation tools, release 12.4, V12.4.99
Build cuda_12.4.r12.4/compiler.33961263_0

安装cudnn

1.下载cudnn

cudnn下载地址

2.安装cudnn
复制代码
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt-get update
sudo apt-get -y install cudnn
#我们这里是cuda12所以安装12
sudo apt-get -y install cudnn-cuda-12
3.查找cudnn安装位置
复制代码
whereis cudnn_version.h
#返回下面
root@ef174907636d:~# whereis cudnn_version.h
cudnn_version.h: /usr/include/cudnn_version.h
4.将cudnn相关安装文件cp到cuda下
复制代码
cp /usr/include/cudnn* /usr/local/cuda-12.4/include
5.检查是否安装成功
复制代码
cat /usr/local/cuda-12.4/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
dpkg -l |grep cudnn

安装paddlepaddle-gpu

复制代码
 python -m pip install paddlepaddle-gpu==2.6.2.post120 -i https://www.paddlepaddle.org.cn/packages/stable/cu120/

安装PaddleOCR

复制代码
# 先卸载掉fastapi,不然会发生版本冲突
pip uninstall fastapi
pip install PaddleOCR   -i https://pypi.tuna.tsinghua.edu.cn/simple
# 再安装fastapi
pip install fastapi

安装相关依赖包

复制代码
apt-get install libgl1-mesa-glx
相关推荐
维度攻城狮2 小时前
实现在Unity3D中仿真汽车,而且还能使用ros2控制
python·unity·docker·汽车·ros2·rviz2
简简单单做算法2 小时前
基于mediapipe深度学习和限定半径最近邻分类树算法的人体摔倒检测系统python源码
人工智能·python·深度学习·算法·分类·mediapipe·限定半径最近邻分类树
就决定是你啦!3 小时前
机器学习 第一章 绪论
人工智能·深度学习·机器学习
有个人神神叨叨5 小时前
OpenAI发布的《Addendum to GPT-4o System Card: Native image generation》文件的详尽笔记
人工智能·笔记
林九生5 小时前
【Python】Browser-Use:让 AI 替你掌控浏览器,开启智能自动化新时代!
人工智能·python·自动化
liuyunshengsir6 小时前
AI Agent 实战:搭建个人在线旅游助手
人工智能·旅游
Shawn_Shawn6 小时前
大模型微调介绍
人工智能
TiAmo zhang6 小时前
DeepSeek-R1 模型现已在亚马逊云科技上提供
人工智能·云计算·aws
liruiqiang056 小时前
循环神经网络 - 简单循环网络
人工智能·rnn·深度学习·神经网络·机器学习
Erica_zhase6 小时前
GPT-4o推出的原生图像生成功能升级后有点东西!
人工智能