mirror of
https://github.com/csukuangfj/kaldifeat.git
synced 2025-08-09 01:52:39 +00:00
77 lines
1.9 KiB
Bash
Executable File
77 lines
1.9 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
#
|
|
set -ex
|
|
|
|
if [ -z $PYTHON_VERSION ]; then
|
|
echo "Please set the environment variable PYTHON_VERSION"
|
|
echo "Example: export PYTHON_VERSION=3.8"
|
|
# Valid values: 3.8, 3.9, 3.10, 3.11
|
|
exit 1
|
|
fi
|
|
|
|
if [ -z $TORCH_VERSION ]; then
|
|
echo "Please set the environment variable TORCH_VERSION"
|
|
echo "Example: export TORCH_VERSION=1.10.0"
|
|
exit 1
|
|
fi
|
|
|
|
export PATH=$PYTHON_INSTALL_DIR/bin:$PATH
|
|
export LD_LIBRARY_PATH=$PYTHON_INSTALL_DIR/lib:$LD_LIBRARY_PATH
|
|
ls -lh $PYTHON_INSTALL_DIR/lib/
|
|
|
|
python3 --version
|
|
which python3
|
|
|
|
python3 -m pip install scikit-build
|
|
python3 -m pip install -U pip cmake
|
|
python3 -m pip install wheel twine typing_extensions
|
|
python3 -m pip install bs4 requests tqdm auditwheel
|
|
|
|
echo "Installing torch"
|
|
python3 -m pip install -qq torch==$TORCH_VERSION+cpu -f https://download.pytorch.org/whl/torch_stable.html || \
|
|
python3 -m pip install -qq torch==$TORCH_VERSION+cpu -f https://download.pytorch.org/whl/torch/
|
|
|
|
rm -rf ~/.cache/pip
|
|
yum clean all
|
|
|
|
cd /var/www
|
|
|
|
export CMAKE_CUDA_COMPILER_LAUNCHER=
|
|
export KALDIFEAT_CMAKE_ARGS=" -DPYTHON_EXECUTABLE=$PYTHON_INSTALL_DIR/bin/python3 "
|
|
export KALDIFEAT_MAKE_ARGS=" -j "
|
|
|
|
nvcc --version || true
|
|
rm -rf /usr/local/cuda*
|
|
nvcc --version || true
|
|
|
|
python3 setup.py bdist_wheel
|
|
if [[ x"$IS_2_28" == x"1" ]]; then
|
|
plat=manylinux_2_28_x86_64
|
|
else
|
|
plat=manylinux_2_17_x86_64
|
|
fi
|
|
|
|
auditwheel --verbose repair \
|
|
--exclude libc10.so \
|
|
--exclude libc10_cuda.so \
|
|
--exclude libcuda.so.1 \
|
|
--exclude libcudart.so.${CUDA_VERSION} \
|
|
--exclude libnvToolsExt.so.1 \
|
|
--exclude libnvrtc.so.${CUDA_VERSION} \
|
|
--exclude libtorch.so \
|
|
--exclude libtorch_cpu.so \
|
|
--exclude libtorch_cuda.so \
|
|
--exclude libtorch_python.so \
|
|
\
|
|
--exclude libcudnn.so.8 \
|
|
--exclude libcublas.so.11 \
|
|
--exclude libcublasLt.so.11 \
|
|
--exclude libcudart.so.11.0 \
|
|
--exclude libnvrtc.so.11.2 \
|
|
--exclude libtorch_cuda_cu.so \
|
|
--exclude libtorch_cuda_cpp.so \
|
|
--plat $plat \
|
|
dist/*.whl
|
|
|
|
ls -lh /var/www
|