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 33369c7 commit bbdf3c4Copy full SHA for bbdf3c4
.github/workflows/cuda/cu111-Windows-env.sh
@@ -5,4 +5,4 @@ PATH=${CUDA_HOME}/bin:$PATH
5
PATH=/c/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio/2017/BuildTools/MSBuild/15.0/Bin:$PATH
6
7
export FORCE_CUDA=1
8
-export TORCH_CUDA_ARCH_LIST="6.0+PTX"
+export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5"
0 commit comments