You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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。The text was updated successfully, but these errors were encountered: