feat: 激活torchaudio的环境

build_space/Dockerfile.jupyterlab_ubuntu

@@ -49,7 +49,7 @@ RUN cp /etc/apt/sources.list /etc/apt/sources.list.bak \
     deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ ${version_id}-proposed main restricted universe multiverse\n\
     deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ ${version_id}-backports main restricted universe multiverse" > /etc/apt/sources.list \
     && apt-get update -y \
-    && apt-get install --no-install-recommends -y vim openssl libssl-dev net-tools locales wget git git-lfs sudo openssh-client openssh-server \
+    && apt-get install --no-install-recommends -y vim openssl libssl-dev net-tools locales wget git git-lfs sudo openssh-client openssh-server libaio-dev \
     && locale-gen en_US.UTF-8 
 
 # ----- conda and python install -----
@@ -171,6 +171,19 @@ RUN jupytersite="$(python3 -m pip show jupyterlab | grep -i '^location' | awk '{
     && ssh-keygen  -A \
     && sed -i "s/#UseDNS .*/UseDNS no/" /etc/ssh/sshd_config
 
+# RUN echo "source /usr/local/bin/fastpt -E  > /dev/null 2>&1" >> /etc/bash.bashrc
+
+RUN /bin/bash -c "source /usr/local/bin/fastpt -E" && echo 'source /etc/bashrc' >> /etc/profile.d/custom.sh
+
+ENV FASTPT_TORCH_LIB_PATH=/usr/local/lib/python3.10/dist-packages/torch/lib
+
+ENV FASTPT_CUDA_LIB_PATH=/opt/dtk/cuda/cuda-12/lib64
+
+ENV LD_LIBRARY_PATH=$FASTPT_TORCH_LIB_PATH:$LD_LIBRARY_PATH
+
+ENV LD_LIBRARY_PATH=$FASTPT_CUDA_LIB_PATH:$LD_LIBRARY_PATH
+
+
 EXPOSE 8888