File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -58,7 +58,7 @@ for whl_file in "$@"; do
58
58
rm -rf " ${whl_dir} " /torch/lib/libnvrtc*
59
59
(find " ${whl_dir} /torch/include/caffe2" -maxdepth 1 -mindepth 1 -type d| grep -v serialize| xargs rm -rf) || echo " caffe2 is empty already"
60
60
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
61
+ 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
62
63
63
sed -i -e " s/-with-pypi-cudnn//g" " ${whl_dir} /torch/version.py"
64
64
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