torchvision报错
收藏回复举报
torchvision报错
t('forum.solved') 已解决
发表于2023-04-06 16:05:28
0 查看

在华为atlas200 ai加速模块  CANN6.0.0 运行深度学习模块  yolov3的 non_max_suppression函数 出现以下报错,我的torch和 torchvision版本是一致的,是不是在atlas下面不能用torchvision?

File "/root/archiconda3/envs/th_npu/lib/python3.7/site-packages/torchvision/extension.py", line 63, in _assert_has_ops 

    "Couldn't load custom C++ ops. This can happen if your PyTorch and " 

RuntimeError: Couldn't load custom C++ ops. This can happen if your PyTorch and torchvision versions are incompatible, or if you had errors while compiling torchvision from source. For further information on the compatible versions, check https://github.com/pytorch/vision#installation for the compatibility matrix. Please check your PyTorch version with torch.__version__ and your torchvision version with torchvision.__version__ and verify if they are compatible, and if not please reinstall torchvision so that it matches your PyTorch install.

cke_383.png

我要发帖子