diff --git a/toolchain/build_abacus_gnu.sh b/toolchain/build_abacus_gnu.sh
index bff9620cab..45f4a2995e 100755
--- a/toolchain/build_abacus_gnu.sh
+++ b/toolchain/build_abacus_gnu.sh
@@ -29,8 +29,8 @@ FFTW3=$INSTALL_DIR/fftw-3.3.10
 CEREAL=$INSTALL_DIR/cereal-1.3.2/include/cereal
 LIBXC=$INSTALL_DIR/libxc-6.2.2
 RAPIDJSON=$INSTALL_DIR/rapidjson-1.1.0/
-# LIBRI=$INSTALL_DIR/LibRI-0.1.1
-# LIBCOMM=$INSTALL_DIR/LibComm-0.1.0
+# LIBRI=$INSTALL_DIR/LibRI-0.2.1.0
+# LIBCOMM=$INSTALL_DIR/LibComm-0.1.1
 # LIBTORCH=$INSTALL_DIR/libtorch-2.1.2/share/cmake/Torch
 # LIBNPY=$INSTALL_DIR/libnpy-1.0.1/include
 # DEEPMD=$HOME/apps/anaconda3/envs/deepmd
diff --git a/toolchain/build_abacus_intel-mpich.sh b/toolchain/build_abacus_intel-mpich.sh
index 5669d071aa..40456c3df8 100755
--- a/toolchain/build_abacus_intel-mpich.sh
+++ b/toolchain/build_abacus_intel-mpich.sh
@@ -29,8 +29,8 @@ LIBXC=$INSTALL_DIR/libxc-6.2.2
 RAPIDJSON=$INSTALL_DIR/rapidjson-1.1.0/
 # LIBTORCH=$INSTALL_DIR/libtorch-2.1.2/share/cmake/Torch
 # LIBNPY=$INSTALL_DIR/libnpy-1.0.1/include
-# LIBRI=$INSTALL_DIR/LibRI-0.1.1
-# LIBCOMM=$INSTALL_DIR/LibComm-0.1.0
+# LIBRI=$INSTALL_DIR/LibRI-0.2.1.0
+# LIBCOMM=$INSTALL_DIR/LibComm-0.1.1
 # DEEPMD=$HOME/apps/anaconda3/envs/deepmd
 
 cmake -B $BUILD_DIR -DCMAKE_INSTALL_PREFIX=$PREFIX \
diff --git a/toolchain/build_abacus_intel.sh b/toolchain/build_abacus_intel.sh
index 90e5c86a2b..7df31a3b55 100755
--- a/toolchain/build_abacus_intel.sh
+++ b/toolchain/build_abacus_intel.sh
@@ -29,8 +29,8 @@ LIBXC=$INSTALL_DIR/libxc-6.2.2
 RAPIDJSON=$INSTALL_DIR/rapidjson-1.1.0/
 # LIBTORCH=$INSTALL_DIR/libtorch-2.1.2/share/cmake/Torch
 # LIBNPY=$INSTALL_DIR/libnpy-1.0.1/include
-# LIBRI=$INSTALL_DIR/LibRI-0.1.1
-# LIBCOMM=$INSTALL_DIR/LibComm-0.1.0
+# LIBRI=$INSTALL_DIR/LibRI-0.2.1.0
+# LIBCOMM=$INSTALL_DIR/LibComm-0.1.1
 # DEEPMD=$HOME/apps/anaconda3/envs/deepmd
 
 # if use deepks and deepmd
diff --git a/toolchain/scripts/stage4/install_libtorch.sh b/toolchain/scripts/stage4/install_libtorch.sh
index b29fd7789e..5320b27539 100755
--- a/toolchain/scripts/stage4/install_libtorch.sh
+++ b/toolchain/scripts/stage4/install_libtorch.sh
@@ -37,17 +37,17 @@ case "${with_libtorch}" in
   __INSTALL__)
     echo "==================== Installing libtorch ===================="
     dirname="libtorch-${libtorch_ver}"
-    filename="${dirname}.zip"
-    pkg_install_dir="${INSTALLDIR}/${filename}"
+    pkg_install_dir="${INSTALLDIR}/${dirname}"
     #pkg_install_dir="${HOME}/lib/libtorch/${libtorch_ver}"
     install_lock_file="${pkg_install_dir}/install_successful"
     archive_file="libtorch-cxx11-abi-shared-with-deps-${libtorch_ver}%2Bcpu.zip"
+    filename="${dirname}.zip"
 
     if verify_checksums "${install_lock_file}"; then
       echo "${filename} is already installed, skipping it."
     else
-        if [ -f ${archive_file} ]; then
-            echo "${archive_file} is found"
+        if [ -f ${filename} ]; then
+            echo "${filename} is found"
         else
             # download from pytorch.com and checksum
             url=https://download.pytorch.org/libtorch/cpu/${archive_file}