项目场景:
insigthface这个项目真的有点费劲,研究了两三天,到处跳坑,刚总算是能够在公司设备上运行起来了,这里暂时先记录下自己这两天遇到的坑。
先说下我这边的设备配置:
显卡型号:GeForce RTX 2080 SUPER
发行版本:CentOS Linux 7 (Core)
最开始的CUDA版本:9.0(**跳坑的一切罪恶之源 ** )
问题描述:
在以上配置之下,当我按照官网源码步骤想测试一下的时候,我输入以下运行命令
python recognition/ArcFace/verification.py
然后就出现了下面的错误
raise MXNetError(py_str(_LIB.MXGetLastError()))
mxnet.base.MXNetError: [10:10:32] src/operator/fusion/fused_op.cu:604: Check failed: compileResult == NVRTC_SUCCESS (5 vs. 0) : NVRTC Compilation failed.
Please set environment variable MXNET_USE_FUSION to 0.
nvrtc: error: invalid value for --gpu-architecture (-arch)
当看到这个提示时我以为需要设置下变量就可,
于是输入:
export MXNET_USE_FUSION=0
输出如下:
loading /home/user/Desktop/insightface-master/models/model-r100-ii/model 0
[10:16:41] src/nnvm/legacy_json_util.cc:209: Loading symbol saved by previous version v1.2.0. Attempting to upgrade...
[10:16:41] src/nnvm/legacy_json_util.cc:217: Symbol successfully upgraded!
[10:16:41] src/base.cc:51: Upgrade advisory: this mxnet has been built against cuda library version 9000, which is older than the oldest version tested by CI (10000).
Set MXNET_CUDA_LIB_CHECKING=0 to quiet this warning