revert packaging C library - not compatible

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Jan 23, 2024 · 486cca4 · 486cca4
1 parent 423e5c4
commit 486cca4
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 9 deletions.
diff --git a/source/install/docker_package_c.sh b/source/install/docker_package_c.sh
@@ -2,15 +2,10 @@ set -e
 
 SCRIPT_PATH=$(dirname $(realpath -s $0))
 
-wget https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-2.1.2%2Bcpu.zip -O libtorch.zip
-unzip libtorch.zip
-mv libtorch ${SCRIPT_PATH}/../libtorch
-
 docker run --rm -v ${SCRIPT_PATH}/../..:/root/deepmd-kit -w /root/deepmd-kit \
 	tensorflow/build:${TENSORFLOW_BUILD_VERSION:-2.15}-python3.11 \
 	/bin/sh -c "pip install \"tensorflow${TENSORFLOW_VERSION}\" cmake \
             && cd /root/deepmd-kit/source/install \
             && CC=/dt9/usr/bin/gcc \
                CXX=/dt9/usr/bin/g++ \
-               CMAKE_PREFIX_PATH=/root/deepmd-kit/source/libtorch \
                /bin/sh package_c.sh"
diff --git a/source/install/package_c.sh b/source/install/package_c.sh
@@ -15,10 +15,7 @@ NPROC=$(nproc --all)
 BUILD_TMP_DIR=${SCRIPT_PATH}/../build_c
 mkdir -p ${BUILD_TMP_DIR}
 cd ${BUILD_TMP_DIR}
-cmake \
-	-D ENABLE_TENSORFLOW=ON \
-	-D ENABLE_PYTORCH=ON \
-	-D CMAKE_INSTALL_PREFIX=${INSTALL_PREFIX} \
+cmake -DCMAKE_INSTALL_PREFIX=${INSTALL_PREFIX} \
 	-DUSE_CUDA_TOOLKIT=TRUE \
 	-DPACKAGE_C=TRUE \
 	-DUSE_TF_PYTHON_LIBS=TRUE \