We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent fa6e81d commit 7e10fc8Copy full SHA for 7e10fc8
.github/workflows/release.yml
@@ -128,7 +128,7 @@ jobs:
128
run: |
129
export PATH=/usr/local/cuda/bin:$PATH
130
export FORCE_CUDA=1
131
- export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
+ export TORCH_CUDA_ARCH_LIST="6.0;7.0;7.5;8.0;8.6+PTX"
132
TORCH_VERSION=`echo "pt${{ matrix.include-combo.torch-version }}" | sed "s/..$//" | sed "s/\.//g"`
133
CUDA_VERSION=`echo ${{ matrix.include-combo.cuda-version }}`
134
export TORCHSORT_VERSION_SUFFIX="+$TORCH_VERSION$CUDA_VERSION"
0 commit comments