diff --git a/finetune/Dockerfile b/finetune/Dockerfile
index a1a3c46..97aec6d 100644
--- a/finetune/Dockerfile
+++ b/finetune/Dockerfile
@@ -356,6 +356,8 @@ EOT
 
 RUN <<EOT
 #!/bin/bash
+source /opt/conda/etc/profile.d/conda.sh
+conda activate ${CONDA_ENV_NAME}
 pip install optimum
 pip install peft tiktoken \
     tqdm matplotlib seaborn numpy pandas scikit-learn diffusers \