error: command ‘C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.1\\bin\\nvcc.exe‘ failed

项目场景:

在win10(CUDA10.1, pytorch=1.8.1)第三方库Detectron2(version==0.6)安装后进行编译时

python setup.py build

报错的第一条信息为:

cpp_extension.py:335: UserWarning: Attempted to use ninja as the BuildExtension backend but we could not find ninja.. Falling back to using the slow distutils backend.
      warnings.warn(msg.format('we could not find ninja.'))

问题解决:

首先输入

pip install ninja

随后依次输入

python setup.py build
python setup.py install

即可
输入

conda list

查看已经安装的库
error: command ‘C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.1\\bin\\nvcc.exe‘ failed_第1张图片
输入

import detectron2

没有报错,解决问题
在这里插入图片描述

原因分析:

这个问题歪打正着解决了,但具体原因疑似是windows上没有合适的编译系统产生的?(存疑)
其他的可能解决方法:可能是因为在torch1.5以后不再使用AT_CHECK,而是TORCH_CHECK。只需要将报错文件的AT_CHECK改为TORCH_CHECK即可。
相关的链接:
https://blog.csdn.net/u011058765/article/details/117000663

https://blog.csdn.net/XUDINGYI312/article/details/120742917

你可能感兴趣的:(c语言,pytorch,深度学习)