We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bb84c0b commit 32ebc48Copy full SHA for 32ebc48
release/pypi/prep_binary_for_pypi.sh
@@ -57,6 +57,7 @@ for whl_file in "$@"; do
57
rm -rf "${whl_dir}/caffe2"
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
+ 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
62
63
sed -i -e "s/-with-pypi-cudnn//g" "${whl_dir}/torch/version.py"
0 commit comments