Skip to content

Commit

Permalink
[ci skip]
Browse files Browse the repository at this point in the history
  • Loading branch information
@rusty1s
rusty1s committed Sep 9, 2021
1 parent d168bf7 commit f31fa47
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion .github/workflows/cuda/cu111-Windows-env.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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"

0 comments on commit f31fa47

Please sign in to comment.