提交 5a378324 编写于 作者: M Megvii Engine Team

fix(whl/misc): easy whl local test, and fix export ALL_PYTHON at docker

env when value include more than two python version

GitOrigin-RevId: 8f1c0eb419970831be81d7b426c168ede4ee0e3f
上级 fc3eca84
...@@ -70,12 +70,12 @@ ...@@ -70,12 +70,12 @@
If you just want to build for a specific Python verison, you can use `ALL_PYTHON` environment variable. eg: If you just want to build for a specific Python verison, you can use `ALL_PYTHON` environment variable. eg:
``` ```
ALL_PYTHON=35m ./scripts/whl/manylinux2010/build_wheel.sh ALL_PYTHON="36m" ./scripts/whl/manylinux2010/build_wheel.sh
``` ```
If you just want to build with cpu only version, you can set `BUILD_WHL_CPU_ONLY` environment 'ON'. eg: If you just want to build with cpu only version, you can set `BUILD_WHL_CPU_ONLY` environment 'ON'. eg:
``` ```
BUILD_WHL_CPU_ONLY=ON ALL_PYTHON=35m ./scripts/whl/manylinux2010/build_wheel.sh BUILD_WHL_CPU_ONLY="ON" ALL_PYTHON="36m" ./scripts/whl/manylinux2010/build_wheel.sh
``` ```
## build for macos ## build for macos
...@@ -85,7 +85,7 @@ ...@@ -85,7 +85,7 @@
If you just want to build for a specific Python verison, you can use `ALL_PYTHON` environment variable. eg: If you just want to build for a specific Python verison, you can use `ALL_PYTHON` environment variable. eg:
``` ```
ALL_PYTHON=3.5.9 ./scripts/whl/macos/macos_build_whl.sh ALL_PYTHON="3.7.7" ./scripts/whl/macos/macos_build_whl.sh
``` ```
## build for windows ## build for windows
``` ```
...@@ -94,10 +94,10 @@ ...@@ -94,10 +94,10 @@
If you just want to build for a specific Python verison, you can use `ALL_PYTHON` environment variable. eg: If you just want to build for a specific Python verison, you can use `ALL_PYTHON` environment variable. eg:
``` ```
ALL_PYTHON=3.5.4 ./scripts/whl/windows/windows_build_whl.sh ALL_PYTHON="3.8.3" ./scripts/whl/windows/windows_build_whl.sh
``` ```
If you just want to build with cpu only version, you can set `BUILD_WHL_CPU_ONLY` environment 'ON'. eg: If you just want to build with cpu only version, you can set `BUILD_WHL_CPU_ONLY` environment 'ON'. eg:
``` ```
BUILD_WHL_CPU_ONLY='ON' ALL_PYTHON=3.5.4 ./scripts/whl/windows/windows_build_whl.sh BUILD_WHL_CPU_ONLY="ON" ALL_PYTHON="3.8.3" ./scripts/whl/windows/windows_build_whl.sh
``` ```
...@@ -100,4 +100,4 @@ if [ ${BUILD_WHL_CPU_ONLY} = "OFF" ]; then ...@@ -100,4 +100,4 @@ if [ ${BUILD_WHL_CPU_ONLY} = "OFF" ]; then
fi fi
fi fi
docker run -it --rm $TMPFS_ARGS -e UID=${USERID} -e LOCAL_VERSION=${LOCAL_VERSION} -e BUILD_WHL_CPU_ONLY=${BUILD_WHL_CPU_ONLY} -e ALL_PYTHON=${ALL_PYTHON} -v ${CUDA_ROOT_DIR}:/usr/local/cuda -v ${CUDNN_ROOT_DIR}:/opt/cudnn -v ${TENSORRT_ROOT_DIR}:/opt/tensorrt -v ${BASEDIR}:/home/code -v ${OUTPUTDIR}:/home/output:rw env_manylinux2010:latest /home/code/scripts/whl/manylinux2010/do_build.sh docker run -it --rm $TMPFS_ARGS -e UID=${USERID} -e LOCAL_VERSION=${LOCAL_VERSION} -e BUILD_WHL_CPU_ONLY=${BUILD_WHL_CPU_ONLY} -e ALL_PYTHON="${ALL_PYTHON}" -v ${CUDA_ROOT_DIR}:/usr/local/cuda -v ${CUDNN_ROOT_DIR}:/opt/cudnn -v ${TENSORRT_ROOT_DIR}:/opt/tensorrt -v ${BASEDIR}:/home/code -v ${OUTPUTDIR}:/home/output:rw env_manylinux2010:latest /home/code/scripts/whl/manylinux2010/do_build.sh
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册