报错代码:
Loading extension module total_variation_cuda...
Traceback (most recent call last):
File "run.py", line 915, in <module>
cfg = mmcv.Config.fromfile(args.config)
AttributeError: module 'mmcv' has no attribute 'Config'
原因分析:mmcv最新模块中已不再支持Config类,直接安装mmengine即可,将原先的import mmcv改成import mmengine。详见https://github.com/open-mmlab/mmdeploy/issues/1781。还有最后需要在命令行中输入以下命令:
export CUDA_HOME=/usr/local/cuda-11.7