Xavier及Orin安装torch和torchvision后，在运行YOLOv5时出现的版本不兼容问题

最新推荐文章于 2023-09-05 15:54:05 发布

老干饭拌妈

最新推荐文章于 2023-09-05 15:54:05 发布

阅读量159

点赞数

文章标签： pytorch

本文链接：https://blog.csdn.net/weixin_48351638/article/details/131156489

版权

按照NVIDIA官网指示，在arm64上安装torch和torchvision后，运行YOLOv5出现以下问题：

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.

解决方法参考以下链接，其中提问者自己找到的解决方法对我有效：

https://forums.developer.nvidia.com/t/pytorch-and-torchvision-versions-are-incompatible/231225/4