Skip to content

Commit f52bc92

Browse files
authored
Changes to prep for pypi script for release 1.13.1 (#1231)
1 parent d034c3c commit f52bc92

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

release/pypi/prep_binary_for_pypi.sh

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,9 +56,8 @@ for whl_file in "$@"; do
5656
if [[ $whl_file == *"with.pypi.cudnn"* ]]; then
5757
rm -rf "${whl_dir}/caffe2"
5858
rm -rf "${whl_dir}"/torch/lib/libnvrtc*
59-
find "${whl_dir}/torch/include/caffe2" -maxdepth 1 -mindepth 1 -type d|grep -v serialize|xargs rm -rf
6059
sed -i -e "s/Requires-Dist: nvidia-cuda-runtime-cu11/Requires-Dist: nvidia-cuda-runtime-cu11 (==11.7.99)/" "${whl_dir}"/*/METADATA
61-
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
60+
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
6261

6362
sed -i -e "s/-with-pypi-cudnn//g" "${whl_dir}/torch/version.py"
6463
find "${whl_dir}/torch/" -maxdepth 1 -type f -name "*.so*" | while read sofile; do

0 commit comments

Comments
 (0)