feat: 解决tensorflow-text没有版本匹配；找不到nvcc的问题

build_space/Dockerfile.jupyterlab_ubuntu

@@ -62,15 +62,6 @@ RUN if [ -n "$CONDA_URL" ];then \
     && cd .. \
     && rm -rf /tmp/conda-extension; fi
 
-#RUN if [ $BASE_IMAGE_IS_TORCH -eq 0 ];then \
-#    mkdir -p /tmp/conda-extension \
-#    && cd /tmp/conda-extension \
-#    && wget $CONDA_URL \
-#    && bash $(echo $CONDA_URL | awk -F "/" '{print $NF}') -b -p /opt/conda \
-#    && echo "export PATH=\$PATH:/opt/conda/bin" >> /etc/profile.d/sothisai.sh \
-#    && cd .. \
-#    && rm -rf /tmp/conda-extension; fi
-
 ENV PATH=$PATH:/opt/conda/bin
 
 RUN pip3 install --upgrade pip ${SOURCES} || pip install --upgrade pip ${SOURCES} \
@@ -87,11 +78,14 @@ RUN if [ -n "$TORCH_VERSION" ];then \
     pip install --no-cache-dir transformers accelerate diffusers; fi
 
 RUN if [ -n "$TENSORFLOW_VERSION" ]; then \
-    tf_version_minor=$(echo $TENSORFLOW_VERSION | cut -d'.' -f1-2 ) && \
+    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==$tf_version_minor.* tf-models-official==$tf_version_minor.* && \
+    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; fi
+    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
 
 # ----- paddlepaddle install -----
 RUN if [ -n "$PADDLEPADDLE_VERSION" ] && [ -n "$PADDLE_URL" ]; then \