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 77f6ba5 commit f319a87Copy full SHA for f319a87
.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="5.0+PTX;6.0;7.0;7.5;8.0;8.6"
+export TORCH_CUDA_ARCH_LIST="6.0+PTX;7.0;7.5;8.0;8.6"
setup.py
@@ -33,9 +33,6 @@ def get_extensions():
33
extra_compile_args = {'cxx': ['-O2']}
34
extra_link_args = ['-s']
35
36
- if sys.platform == 'win32':
37
- extra_compile_args['cxx'].append('/Zm500')
38
-
39
info = parallel_info()
40
if ('backend: OpenMP' in info and 'OpenMP not found' not in info
41
and sys.platform != 'darwin'):
0 commit comments