PyTorch: cpp_extension.py:184: предупреждение пользователя: ошибка проверки версии компилятора

Question

У меня проблема с pytorch:

, когда я запускаю python setup.py build_ext --inplace

Я получаю:

... работает build_ext C: \python36 \ lib \ site-packages \ torch \ utils \ cpp_extension.py: 184: UserWarning: Ошибка при проверке версии компилятора для cl: [WinError 2] Das System kann die angegebene Datei nicht finden warnings.warn ('Ошибка проверки версии компилятора для {}: {} '. формат (компилятор, ошибка)) сборка' pointnet2._ext 'расширение

НАСТРОЙКА:

CUDA 9.0

C: \ Program Files (x86)\ Microsoft Visual Studio 14.0 \ VC \ bin> nvcc --version nvcc: драйвер компилятора NVIDIA® Cuda Copyright (c) 2005-2017 NVIDIA Corporation Построен на Fri_Sep__1_21: 08: 32_Central_Daylight_Time_2017 Инструменты компиляции Cuda, выпуск 9.0, V9.0.176

MSVC

C: \ Program Files (x86) \ Microsoft Visual Studio 14.0 \ VC \ bin> cl Microsoft (R) C / C ++ Оптимизирующая версия компилятора 19.00.24215.1 для x86 Copyright (C)Корпорация Майкрософт.Все права защищены.

Pytorch 1.0.1 (установлен через PIP)

Additonal:

Когда я запускаю:

python setup.py install

Я получаю:

...

  File "C:\python36\lib\distutils\command\build_ext.py", line 448, in build_extensions
    self._build_extensions_serial()
  File "C:\python36\lib\distutils\command\build_ext.py", line 473, in _build_extensions_serial
    self.build_extension(ext)
  File "C:\python36\lib\site-packages\setuptools\command\build_ext.py", line 199, in build_extension
    _build_ext.build_extension(self, ext)
  File "C:\python36\lib\distutils\command\build_ext.py", line 558, in build_extension
    target_lang=language)
  File "C:\python36\lib\distutils\ccompiler.py", line 717, in link_shared_object
    extra_preargs, extra_postargs, build_temp, target_lang)
  File "C:\python36\lib\distutils\_msvccompiler.py", line 501, in link
    build_temp = os.path.dirname(objects[0])
IndexError: list index out of range

setup.py

setup(
...  
install_requires=requirements,
    ext_modules=[
        CUDAExtension(
            name="pointnet2._ext",
            sources=_ext_sources,
            extra_compile_args={
                "cxx": ["-O2", "-I{}".format("{}/include".format(_ext_src_root))],
                "nvcc": ["-O2", "-I{}".format("{}/include".format(_ext_src_root))],
            },
        )
    ],
...
)