1.问题描述

Couldn‘t load custom C++ ops. This can happen if your PyTorch and torchvision versions are incompatible.
第一次遇到这个问题，说是torch和torchvision版本不对应。但是这个环境我已经使用好久了，一直没有出现问题（用conda安装的）。但是既然问题出现了，就得想办法解决！

2.分析问题

2.1 查看torch和torchvision版本

可以使用以下代码在Pycharm中查看

import torch
print(torch.__version__)
import torchvision
print(torchvision.__version__)
print(torch.version.cuda)

输出：

1.12.1
0.13.1
11.3

看了一些torch和torchvision的版本独对应关系，发现没啥问题torch1.12.1应该是对应torchvision0.13.1，貌似没啥问题。

2.2 重新安装torch，torchvision，torchaudio

结合网上一些说法（说法一：把torch，torchvision，torchaudion卸载干净重新安装对应的版本；说法二：推荐使用pip安装），因此，我先将torch，torchvision，torchaudion全部卸载，然后重新用pip安装。

2.2.1卸载torch，torchvision，torchaudion

pip uninstall torch
pip uninstall torchvision
pip uninstall torchaudio

conda uninstall torch
conda uninstall torchvision
conda uninstall torchaudio

这下总下载干净了

2.2.2pip安装

在pytorch官网找到与自己cuda对应的torch，torchvision，torchaudio版本，并且选择用pip安装
我使用的是cuda11.3，pip安装命令如下：

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

安装成功后可以查看下版本

import torch
print(torch.__version__)
import torchvision
print(torchvision.__version__)
print(torch.version.cuda)

输出：

1.12.1+cu113
0.13.1+cu113
11.3

貌似跟刚开始的有点区别，但感觉区别不是很大。
反正最终问题解决了，没有报错Couldn’t load custom C++ ops.

但是又报错显示找不到PIL模块，不过这是个小问题，卸载重装一遍即可。

pip uninstall pillow
pip install pillow

代码正常运行，没有问题了！

3.总结

所以，pip安装就没有问题，conda安装有问题，推荐pip安装。

Couldn’t load custom C++ ops.相关推荐

Win10+yolov5 踩坑记录
Windows10下配置yolov5环境踩坑记录起因随笔基本的环境要求 yolov5各代码作用踩坑的Package windwos下torch和torchvision的安装 windows下p ...
YOLOv5错误提示（ torchvision和torch版本不兼容）
错误提示是 torchvision和torch的版本不兼容 File "D:\Programs\conda411\envs\envZhao\lib\site-packages\torchvi ...
PyTorch and torchvision versions are incompatible问题
Couldn't load custom C++ ops. This can happen if your PyTorch and torchvision versions are incompati ...
Jetson AGX Xavier刷机及环境配置
写在前面:近期打算做一下视觉算法部署的内容,正好实验室有几个Xavier控制器,于是拿来用了一下,从零接触开始nvidia的硬件,以此记录学习过程. 一.Jetson AGX Xavier刷机网上教 ...
OpenVINO Inference Engine之custom Layer自定义算法(VPU版本)
OpenVINO不仅支持广泛的已知深度学习架构(Caff,TensorFlow等),还支持用户自定义CNN 算法,拥有良好的可扩展性.要使用用户自定义功能就要使用到Inference Engine K ...
3DSMAX中英文对照大全(从A-Z分类)
A Absolute Mode Transform Type-in绝对坐标方式变换输入 Absolute/Relative Snap Toggle Mode绝对/相对捕捉开关模式 ACIS Optio ...
玩转NVIDIA Jetson AGX Xavier
最近项目应用需要部署到边缘模块,所以就玩起了Xavier.感觉网上的资料不多,所以从头记录一下笔记,方便大家一起学习应用. 玩转NVIDIA Jetson AGX Xavier 1.刷机(安装sdkm ...
NVIDIA jetson tensorrt加速yolov5摄像头检测
link 在使用摄像头直接检测目标时,检测的实时画面还是有点慢,下面是tensorrt加速过程记录. 一.设备 1.设备jetson agx xavier 2.jetpack4.6.1 3.tenso ...
Jetson Nano部署YOLOv5与Tensorrtx加速——（自己走一遍全过程记录）
说在前面搞了一下Jetson nano和YOLOv5,网上的资料大多重复也有许多的坑,在配置过程中摸爬滚打了好几天,出坑后决定写下这份教程供自己备忘. 事先声明,这篇文章的许多内容本身并不是原创,而 ...

Couldn’t load custom C++ ops.