Pytorch 编译cpp、cuda扩展遇到的问题

参考：https://blog.csdn.net/tanmx219/article/details/100829920

问题一：

C:\Users\pc-01\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\utils\cpp_extension.py:287: UserWarning: Error checking compiler version for cl: 'utf-8' codec can't decode byte 0xd3 in position 0: invalid continuation byte
  warnings.warn('Error checking compiler version for {}: {}'.format(compiler, error))

utf-8的问题，找到报错目录cpp_extension.py的对应代码位置：

也就是283行re.search这里出了问题，把原来的：

match = re.search(r'(\d+)\.(\d+)\.(\d+)', compiler_info.decode().strip())

改成

match = re.search(r'(\d+)\.(\d+)\.(\d+)', compiler_info.decode(' gbk').strip())

究其原因可能是visual studio的那个cl，原来输出是英文的，直接utf-8解码没问题，新版更加“人性化”的能输出中文，然后pytorch这里没改还是utf-8…所以让pytorch按gbk解码就好

问题二：

Traceback (most recent call last):
  File "C:\Users\pc-01\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\utils\cpp_extension.py", line 1539, in _run_ninja_build
    env=env)
  File "C:\Users\pc-01\AppData\Local\Programs\Python\Python37\lib\subprocess.py", line 512, in run
    output=stdout, stderr=stderr)
subprocess.CalledProcessError: Command '['ninja', '-v']' returned non-zero exit status 1.

还是这个cpp_extension.py，找到对应位置：

也就是这个_run_ninja_build方法，把下面这个

command = ['ninja', '-v']

改成

command = ['ninja','--v']

或者

command = ['ninja','--version']

好像只有一种改法有效，大概率为后者。这个东西的作用其实单纯就是指定build.ninja文件的第一行输出：