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 fa8b6d6 commit 5c814e2Copy full SHA for 5c814e2
manywheel/build_cuda.sh
@@ -197,6 +197,8 @@ elif [[ $CUDA_VERSION == "11.8" ]]; then
197
export USE_STATIC_CUDNN=0
198
# Try parallelizing nvcc as well
199
export TORCH_NVCC_FLAGS="-Xfatbin -compress-all --threads 2"
200
+ # Bundle ptxas into the wheel, see https://github.com/pytorch/pytorch/pull/119750
201
+ export BUILD_BUNDLE_PTXAS=1
202
203
if [[ -z "$PYTORCH_EXTRA_INSTALL_REQUIREMENTS" ]]; then
204
echo "Bundling with cudnn and cublas."
0 commit comments