From f52bc9256bfb9806e62907f13ea34c5d10fbae52 Mon Sep 17 00:00:00 2001 From: Andrey Talman Date: Wed, 14 Dec 2022 14:54:01 -0500 Subject: [PATCH] Changes to prep for pypi script for release 1.13.1 (#1231) --- release/pypi/prep_binary_for_pypi.sh | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/release/pypi/prep_binary_for_pypi.sh b/release/pypi/prep_binary_for_pypi.sh index 1eca37cac..639d834a6 100755 --- a/release/pypi/prep_binary_for_pypi.sh +++ b/release/pypi/prep_binary_for_pypi.sh @@ -56,9 +56,8 @@ for whl_file in "$@"; do if [[ $whl_file == *"with.pypi.cudnn"* ]]; then rm -rf "${whl_dir}/caffe2" rm -rf "${whl_dir}"/torch/lib/libnvrtc* - find "${whl_dir}/torch/include/caffe2" -maxdepth 1 -mindepth 1 -type d|grep -v serialize|xargs rm -rf sed -i -e "s/Requires-Dist: nvidia-cuda-runtime-cu11/Requires-Dist: nvidia-cuda-runtime-cu11 (==11.7.99)/" "${whl_dir}"/*/METADATA - sed -i -e "/^Requires-Dist: nvidia-cublas-cu11 (==11.10.3.66).*/a Requires-Dist: nvidia-cuda-nvrtc-cu11 (==11.7.99)" "${whl_dir}"/*/METADATA + sed -i -e "/^Requires-Dist: nvidia-cublas-cu11 (==11.10.3.66).*/a Requires-Dist: nvidia-cuda-nvrtc-cu11 (==11.7.99) ; platform_system == \"Linux\"" "${whl_dir}"/*/METADATA sed -i -e "s/-with-pypi-cudnn//g" "${whl_dir}/torch/version.py" find "${whl_dir}/torch/" -maxdepth 1 -type f -name "*.so*" | while read sofile; do