Commit 67624b99 authored by chenpangpang's avatar chenpangpang
Browse files

feat: 暂存,不安装tensorrt

parent fd41db1c
......@@ -80,12 +80,9 @@ RUN if [ -n "$TORCH_VERSION" ];then \
RUN if [ -n "$TENSORFLOW_VERSION" ]; then \
tf_version_minor=$(echo $TENSORFLOW_VERSION | cut -d'.' -f1-2 ); \
[ "$tf_version_minor" == "2.13" ] || [ "$tf_version_minor" == "2.18" ] && tensorflow_text_version=$tf_version_minor.0rc0 || tensorflow_text_version=$tf_version_minor.*; \
pip install --no-cache-dir tensorflow[and-cuda]==$TENSORFLOW_VERSION \
tensorflow-text==$tensorflow_text_version tensorflow-hub && \
apt-get update -y && \
apt-get install --no-install-recommends -y libnvinfer8 libnvjitlink-12-3 libnvjpeg-12-3 libnvinfer-plugin8 && \
cuda_version=$(echo $BASE_IMAGE | awk -F'[:-]' '{print $2}' | awk -F'.' '{print $1"."$2}') && \
echo "export PATH=\$PATH:/usr/local/cuda-/bin/" >> /etc/profile.d/cuda_nvcc.sh; fi
pip install --no-cache-dir tensorflow[and-cuda]==$TENSORFLOW_VERSION tensorflow-text==$tensorflow_text_version tensorflow-hub && \
# pip install --no-cache-dir "numpy<2" ; fi
# ----- paddlepaddle install -----
RUN if [ -n "$PADDLEPADDLE_VERSION" ] && [ -n "$PADDLE_URL" ]; then \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment