Traceback (most recent call last):
File "/home/llh/Projects/MobileAgent/Mobile-Agent-v2/run.py", line 260, in <module>
ocr_detection = pipeline(Tasks.ocr_detection, model='/home/llh/Models/cv_resnet18_ocr-detection-line-level_damo')
File "/home/llh/miniconda3/envs/mobileagent/lib/python3.9/site-packages/modelscope/pipelines/builder.py", line 170, in pipeline
return build_pipeline(cfg, task_name=task)
File "/home/llh/miniconda3/envs/mobileagent/lib/python3.9/site-packages/modelscope/pipelines/builder.py", line 65, in build_pipeline
return build_from_cfg(
File "/home/llh/miniconda3/envs/mobileagent/lib/python3.9/site-packages/modelscope/utils/registry.py", line 215, in build_from_cfg
raise type(e)(f'{obj_cls.__name__}: {e}')
AttributeError: OCRDetectionPipeline: module 'tensorflow._api.v2.compat.v2.__internal__' has no attribute 'register_load_context_function'
现在安装的版本Pytorch=2.4.0,Tensorflow=2.16.1,测试均可使用CUDA但是运行时会报错:
这边感觉应该是Tensorflow版本太高的原因,但是降级Tensorflow会导致CUDA不可用,因此在不降级CUDA的情况下是否有支持的版本?
requirements.txt
中的Tensorflow=2.9.0或之前的issue中的Tensorflow=2.13.0均无法使用CUDA。