File tree Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Original file line number Diff line number Diff line change @@ -56,9 +56,8 @@ for whl_file in "$@"; do
56
56
if [[ $whl_file == * " with.pypi.cudnn" * ]]; then
57
57
rm -rf " ${whl_dir} /caffe2"
58
58
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
60
59
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
62
61
63
62
sed -i -e " s/-with-pypi-cudnn//g" " ${whl_dir} /torch/version.py"
64
63
find " ${whl_dir} /torch/" -maxdepth 1 -type f -name " *.so*" | while read sofile; do
You can’t perform that action at this time.
0 commit comments