diff --git a/install.sh b/install.sh
index 2eda0fe..63be260 100644
--- a/install.sh
+++ b/install.sh
@@ -1,6 +1,21 @@
 #!/bin/bash
+
+# 安装构建工具
+# Install build tools
 conda install -c conda-forge gcc=14
 conda install -c conda-forge gxx
 conda install ffmpeg cmake
+
+# 设置编译环境
+# Set up build environment
+export CMAKE_MAKE_PROGRAM="$CONDA_PREFIX/bin/cmake"
+export CC="$CONDA_PREFIX/bin/gcc"
+export CXX="$CONDA_PREFIX/bin/g++"
+
 conda install pytorch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 pytorch-cuda=11.8 -c pytorch -c nvidia
-pip install -r requirements.txt
+
+# 刷新环境
+# Refresh environment
+hash -r
+
+pip install -r requirements.txt
\ No newline at end of file