How could I configure `setup.py` to use opencv to write pytorch cpp extension?

Hi,

Following this tutorial, I got to know how to write simple cpp extension for pytorch. However, in my extension, I need to use opencv as a third party library. When I have the line #include <opencv2/opencv.hpp> in my code, I got the error message of opencv2/opencv.hpp: No such file or directory. Would you please tell me how to configure the setup.py file to let the extension support opencv ?

2 Likes