2021-06-29

关于FairMOT模型环境搭建时遇到的一个问题
报错:

RuntimeError:NOT compiled with GPU support
2021-06-29_第1张图片
这种错误首先要排查cuda环境是否装对:版本对应,1.cmd后输入nvcc -V,2.C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\demo_suite路径找到bandwidthTest.exe与deviceQuery.exe两个文件都用cmd去打开看看如果都是pass说明没问题。

然后是我在编译dcnv2的时候有个报错但还是编译出来了,但最终也会导致上面这个错误。
2021-06-29_第2张图片
看似通过,实则包含189行那个没注意。
解决办法就是找到这个文件
try:
if sys.platform.startswith(‘linux’):
minimum_required_version = MINIMUM_GCC_VERSION
version = subprocess.check_output([compiler, ‘-dumpfullversion’, ‘-dumpversion’])
version = version.decode().strip().split(’.’)
else:
minimum_required_version = MINIMUM_MSVC_VERSION
compiler_info = subprocess.check_output(compiler, stderr=subprocess.STDOUT)
match = re.search(r’(\d+).(\d+).(\d+)’, compiler_info.decode(‘gbk’).strip())
version = (0, 0, 0) if match is None else match.groups()
except Exception:
_, error, _ = sys.exc_info()
warnings.warn(‘Error checking compiler version for {}: {}’.format(compiler, error))
return False
按照如上在185行中将解码格式改为gbk,因为pytorch默认的utf-8,ok解决!

这个环境之前也搭过一次,回顾一下多半是dcnv2这部分编译的问题
总结一下:
1.cuda与cudnn版本对应(最好先装然后再建环境,之前在github上有个人的runtime报错就是在这个导致的)
2.vs要装对环境,不然也可能出现编译不出来,记得应该是报cl那个啥啥的错
3.torch里utils文件夹啦那个cpp_extension文件改编码格式(如上文)。

你可能感兴趣的:(计算机视觉)