feat: 修改torch安装参数，适配torch1.xx.xx版本

build_space/Dockerfile.jupyterlab_ubuntu

@@ -55,8 +55,13 @@ RUN pip3 install --upgrade pip ${SOURCES} || pip install --upgrade pip ${SOURCES
     && mv /etc/apt/sources.list.d.bak /etc/apt/sources.list.d
 
 RUN if [ $BASE_IMAGE_IS_TORCH -eq 0 ];then \
-    pip3 install torch==$TORCH_VERSION torchvision==$TORCHVISION_VERSION torchaudio==$TORCHAUDIO_VERSION \
-    --index-url https://download.pytorch.org/whl/cu$(echo "$BASE_IMAGE" | awk -F'[:-]' '{n=split($2,a,"."); print a[1] a[2]}') \
+    cuda_version=$(echo "$BASE_IMAGE" | awk -F'[:-]' '{n=split($2,a,"."); print a[1] a[2]}') \
+    && if [ "${TORCH_VERSION%%.*}" -eq 1 ]; then \
+    pip3 install --no-cache-dir numpy==1.26.4 \
+    && pip3 install --no-cache-dir  torch==$TORCH_VERSION+cu$cuda_version torchvision==$TORCHVISION_VERSION+cu$cuda_version torchaudio==$TORCHAUDIO_VERSION \
+    --extra-index-url https://download.pytorch.org/whl/cu$cuda_version;else \
+    pip3 install --no-cache-dir torch==$TORCH_VERSION torchvision==$TORCHVISION_VERSION torchaudio==$TORCHAUDIO_VERSION \
+    --index-url https://download.pytorch.org/whl/cu$cuda_version; fi \
     && rm -r /root/.cache/pip; fi
 
 COPY ./python-requirements.txt /tmp/