diff --git a/finetune/Dockerfile.update b/finetune/Dockerfile.update
index 7895280..6b8941d 100644
--- a/finetune/Dockerfile.update
+++ b/finetune/Dockerfile.update
@@ -36,9 +36,11 @@ mkdir -p ~/.pip
 wget -qO- https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O /tmp/miniconda.sh
 bash /tmp/miniconda.sh -b -p /opt/conda 
 rm /tmp/miniconda.sh 
-conda init bash 
 ln -s /opt/conda/etc/profile.d/conda.sh /etc/profile.d/conda.sh
 echo ". /opt/conda/etc/profile.d/conda.sh" >> ~/.bashrc 
+. /opt/conda/etc/profile.d/conda.sh 
+conda init bash
+conda config --set show_channel_urls true
 # 配置 .condarc 文件
 cat <<EOF > ~/.condarc
 channels:
@@ -146,14 +148,14 @@ cd ${STAGE_DIR}
 rm -rf ${STAGE_DIR}/MLNX_OFED_LINUX-${MLNX_OFED_VERSION}-ubuntu22.04-x86_64*
 EOT
 
-ENV NV_PEER_MEM_VERSION=1.2
+ARG NV_PEER_MEM_VERSION="1.2"
+ENV NV_PEER_MEM_VERSION=${NV_PEER_MEM_VERSION}
 ENV NV_PEER_MEM_TAG=${NV_PEER_MEM_VERSION}-0
 RUN <<EOT
 #!/bin/bash
 source /opt/conda/etc/profile.d/conda.sh
 conda activate ${CONDA_ENV_NAME}
 # install nv_peer_mem
-apt-get install -y --no-install-recommends dkms gcc-12 dctrl-tools fakeroot linux-headers-generic libnvidia-ml1
 mkdir -p ${STAGE_DIR}
 git clone https://github.com/Mellanox/nv_peer_memory.git --branch ${NV_PEER_MEM_TAG} ${STAGE_DIR}/nv_peer_memory
 cd ${STAGE_DIR}/nv_peer_memory
@@ -162,6 +164,7 @@ cd ${STAGE_DIR}
 tar xzf ${STAGE_DIR}/nvidia-peer-memory_${NV_PEER_MEM_VERSION}.orig.tar.gz
 cd ${STAGE_DIR}/nvidia-peer-memory-${NV_PEER_MEM_VERSION}
 apt-get update
+apt --fix-broken install -y
 apt-get install -y dkms
 dpkg-buildpackage -us -uc
 dpkg -i ${STAGE_DIR}/nvidia-peer-memory_${NV_PEER_MEM_TAG}_all.deb