mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-04-23 00:17:25 +08:00
YuBaoku
b7b4fe6a69
* [CI] Fix prebuilt wheel installation and update Docs * [CI] Update Dockerfile.gpu to restrict SM80/86/89/90, CUDA 12.6 and Python 3.10 * Update nvidia_gpu.md * Update nvidia_gpu.md * Revise NVIDIA GPU installation instructions Updated installation instructions for PaddlePaddle and FastDeploy to remove specific CUDA version mentions and clarify support for multiple GPU architectures. --------- Co-authored-by: Jiang-Jia-Jun <163579578+Jiang-Jia-Jun@users.noreply.github.com>
24 lines
818 B
Docker
24 lines
818 B
Docker
FROM ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/fastdeploy-cuda-12.6:tag-base
|
|
ARG PADDLE_VERSION=3.3.1
|
|
ARG FD_VERSION=2.5.0
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
|
|
WORKDIR /workspace
|
|
|
|
RUN echo "ulimit -u unlimited" >> /root/.bashrc
|
|
RUN echo "ulimit -n 65536" >> /root/.bashrc
|
|
|
|
# uninstall existing package
|
|
RUN python -m pip uninstall paddlepaddle-gpu fastdeploy-gpu -y
|
|
|
|
# install paddlepaddle
|
|
RUN python -m pip install --no-cache-dir paddlepaddle-gpu==${PADDLE_VERSION} -i https://www.paddlepaddle.org.cn/packages/stable/cu126/
|
|
|
|
# build and install FastDeploy
|
|
RUN python -m pip install --no-cache-dir fastdeploy-gpu==${FD_VERSION} -i https://www.paddlepaddle.org.cn/packages/stable/cu126/ --extra-index-url https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple
|
|
|
|
ENV http_proxy=""
|
|
ENV https_proxy=""
|
|
ENV no_proxy=""
|