Cannot import `PaddleOCRVL` in google colab - `ImportError: /usr/local/lib/python3.12/dist-packages/torch/lib/libtorch_cuda.so: undefined symbol: ncclCommWindowRegister`
🔎 Search before asking
- [x] I have searched the PaddleOCR Docs and found no similar bug report.
- [x] I have searched the PaddleOCR Issues and found no similar bug report.
- [x] I have searched the PaddleOCR Discussions and found no similar bug report.
🐛 Bug (问题描述)
Traceback (most recent call last):
File "/usr/local/bin/paddleocr", line 5, in <module>
from paddleocr.__main__ import console_entry
File "/usr/local/lib/python3.12/dist-packages/paddleocr/__init__.py", line 15, in <module>
from paddlex.inference.utils.benchmark import benchmark
File "/usr/local/lib/python3.12/dist-packages/paddlex/__init__.py", line 49, in <module>
from .inference import create_pipeline, create_predictor
File "/usr/local/lib/python3.12/dist-packages/paddlex/inference/__init__.py", line 16, in <module>
from .models import create_predictor
File "/usr/local/lib/python3.12/dist-packages/paddlex/inference/models/__init__.py", line 22, in <module>
from ..utils.official_models import official_models
File "/usr/local/lib/python3.12/dist-packages/paddlex/inference/utils/official_models.py", line 25, in <module>
import modelscope
File "/usr/local/lib/python3.12/dist-packages/modelscope/__init__.py", line 5, in <module>
from modelscope.utils.import_utils import (LazyImportModule,
File "/usr/local/lib/python3.12/dist-packages/modelscope/utils/import_utils.py", line 18, in <module>
from modelscope.utils.ast_utils import (INDEX_KEY, MODULE_KEY, REQUIREMENT_KEY,
File "/usr/local/lib/python3.12/dist-packages/modelscope/utils/ast_utils.py", line 24, in <module>
from modelscope.utils.registry import default_group
File "/usr/local/lib/python3.12/dist-packages/modelscope/utils/registry.py", line 11, in <module>
logger = get_logger()
^^^^^^^^^^^^
File "/usr/local/lib/python3.12/dist-packages/modelscope/utils/logger.py", line 49, in get_logger
from modelscope.utils.torch_utils import is_dist, is_master
File "/usr/local/lib/python3.12/dist-packages/modelscope/utils/torch_utils.py", line 14, in <module>
import torch
File "/usr/local/lib/python3.12/dist-packages/torch/__init__.py", line 427, in <module>
from torch._C import * # noqa: F403
^^^^^^^^^^^^^^^^^^^^^^
ImportError: /usr/local/lib/python3.12/dist-packages/torch/lib/libtorch_cuda.so: undefined symbol: ncclCommWindowRegister
🏃♂️ Environment (运行环境)
colab notebook: (With Python 3.12 on T4 GPU)
https://colab.research.google.com/drive/179HqyXNlNW_TfD8-mLrAY4DiZ4zk1ekX?usp=sharing
🌰 Minimal Reproducible Example (最小可复现问题的Demo)
from paddleocr import PaddleOCRVL
With Python 3.12 on T4 GPU.
Related issues:
- https://github.com/PaddlePaddle/PaddleOCR/issues/16712#issuecomment-3434899092
- https://github.com/PaddlePaddle/PaddleOCR/issues/17049#issue-3610956234
Note: it is running in Kaggle but comes with its own new issues https://github.com/PaddlePaddle/PaddleOCR/issues/17206
@Sunting78 anything on this please?
! python -m pip install paddlepaddle-gpu==3.2.2 -i https://www.paddlepaddle.org.cn/packages/stable/cu126/ !pip install "paddleocr[all]" !pip install --force-reinstall torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
This is my setup. It work on colab. You could try it
After running pip install --force-reinstall torch torchvision torchaudio, it works fine for me
Environment: Linux Python3.11 paddlepaddle-gpu3.2.0