diff --git a/.github/workflows/cuda/cu111-Windows-env.sh b/.github/workflows/cuda/cu111-Windows-env.sh index 12ab1dee..0e672a3f 100644 --- a/.github/workflows/cuda/cu111-Windows-env.sh +++ b/.github/workflows/cuda/cu111-Windows-env.sh @@ -5,4 +5,4 @@ PATH=${CUDA_HOME}/bin:$PATH PATH=/c/Program\ Files\ \(x86\)/Microsoft\ Visual\ Studio/2017/BuildTools/MSBuild/15.0/Bin:$PATH export FORCE_CUDA=1 -export TORCH_CUDA_ARCH_LIST="5.0+PTX" +export TORCH_CUDA_ARCH_LIST="6.0+PTX"