Use MSVC 14.0 with CUDA 11.2

cupy · Feb 22, 2021 · 6c098ca · 6c098ca
1 parent 41c69ca
commit 6c098ca
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/cupy_setup_build.py b/cupy_setup_build.py
@@ -1048,6 +1048,11 @@ def _compile_cu(self, sources, output_dir=None, macros=None,
         cc_args = self._get_cc_args(pp_opts, debug, extra_preargs)
         cuda_version = build.get_cuda_version()
         postargs = _nvcc_gencode_options(cuda_version) + ['-O2']
+        if cuda_version >= 11020:
+            # MSVC 14.0 (2015) is deprecated for CUDA 11.2 but we need it
+            # to build CuPy because some Python versions were built using it.
+            # REF: https://wiki.python.org/moin/WindowsCompilers
+            postargs += ['-allow-unsupported-compiler']
         postargs += ['-Xcompiler', '/MD']
         print('NVCC options:', postargs)