Compiling OpenCV with CUDA Support on Windows

-

I have actively avoided doing this for a while, but now I need to run CUDA-dependent OpenCV Code on a Windows Machine.  So it's time to bite the bullet and get it done.

Want to go for a ride?  It will be fun, I promise!

Background

This is being run on a Gigabyte Motherboard with a 10th-Generation i7 CPU, 64GB RAM and Windows 11 (Release).  You want at least 4GB of RAM and an i5 to do the build with any sort of efficiency.

First, get your pre-requisites

You will need:

Install CUDA, Python, Visual Studio and CMake.  Make sure they are in the user path, and that a handler is installed for "sln" files.  This should be the case by default.

As a sanity check, run "nvidia-smi" to make sure you can talk to your GPU.  On Windows, installing CuDNN involves manually copying files into the CUDA directories, so follow the installation instructions carefully.

If possible, do your build on an SSD, because parts of the process are very I/O heavy.

Configure Everything

Yes, you can use the CMake GUI, but you could also put lit bamboo skewers under your toenails.  The command-line is actually easier.

Firstly extract opencv-x.x.x into a directory (E:\src\opencv\opencv-4.5.4 in my case) and then extract opencv_contrib-x.x.x at the same level.  You should have two directories thus:


Now, change into the opencv-x.x.x directory and create a build directory under that.  Call it something useful like "build".  Change  into that directory.

Now here's the magic.  We're going to get CMake to do all the hard work for us with one simple command:

cmake -DWITH_CUDA=YES -DOPENCV_DNN_CUDA=YES -DENABLE_FAST_MATH=YES -DOPENCV_EXTRA_MODULES_PATH=E:\src\opencv\opencv_contrib-x.x.x\modules -DCUDA_FAST_MATH=YES -DCUDA_ARCH_BIN=7.5 ..

All of the above is a single line, so be careful with the copy and paste.  Also note that the CUDA_ARCH_BIN value needs to be set according to the GPU in use.  You can find a list here: https://en.wikipedia.org/wiki/CUDA.

There are also a load more options you could specify depending on your needs.  Please refer to the OpenCV documentation, but the above will get you started.

That will grind away for a while.  Take a few sips of your beverage.  If all goes to plan, you will have a file "OpenCV.sln" in your build directory.

The Build

Now, open the OpenCV.sln file in Visual Studio".  You should simply be able to type "OpenCV.sln" in the command prompt, and Windows will work stuff out for you.  You'll have a solution view resembling the following:

Probably a good idea to select "Release" for your build type since there is not too much benefit to building debug binaries in this case


Now, expand "CMakeTargets in the Solution Explorer and choose "ALL_BUILD"



Click "Build" from the right-click menu and take a long draught from your beverage.  This is going to take quite a while even on a fast machine.  My system took just on 30 minutes to complete.

There's a lot of warnings, but it should build without error.  Progress will look like this:

Installation

You're probably onto your second pot of coffee by now, but we're approaching the finish line, I promise...

Select the INSTALL project and choose "build" from the context menu.  This will take a couple of minutes to run.

Now try importing "cv2" from within the Python interpreter.  If it succeeds, you're done.  If this fails, go back and check that CMake detected Python in the first instance.









Comments

Post a Comment

Popular posts from this blog

The QYT KT-7900D

-
Image
But is it any good? As you might have noticed, I am now a freshly licensed radio amateur .  As such I do need a transceiver or two in order to get on the air.  This is the story of one of my new radios. I have a limited discretionary budget to use for my new hobby, and I have to say from the outset that my lovely wife has been very supportive.  As such, I want to try and get the best value for our money.  One evening I stumbled upon this little item on Banggood for a little over $A150 including shipping.  Covering both 2m and 70cm bands with a claimed 20W output it seemed almost too good to be true.  I bought one. About ten days later a tiny package arrived for me.  At just 98mm x 43mm x 126mm the photos don't convey quite how small it actually is.  The entire case serves as a heatsink so it is heavier than it looks, but certainly not bulky. The standard power cable comes with a cigarette lighter plug.  As I intended to run it from home usin...
Read more

Recording weather with Arduino, Elasticsearch and Kibana

-
What do half a Bunnings weather station, an Arduino, a NodeMCU (or 3), Elasticsearch, a fishtank and swimming pool have to do with each other? Nothing unless you like to monitor and measure pretty much everything around the house.  I do, and ever since I learned to fly, I have had a fascination with the weather (I am told boaties suffer from a related affliction). I could just buy a weather station, but where's the fun in that when for the same amount of money (OK, maybe a little more) I can make my own and bore people senseless when I go on and on about it? Actually, there is a practical purpose to this as well.  I wanted to teach myself about Elastic and Kibana.  Now you can wade through dry tutorials, but nothing in my opinion beats working with real live data.  Especially if the data is of interest to you. The design What do you want to measure? This is actually pretty important (not to mention obvious), but before embarking too far; take a...
Read more

Exploring Solar Power - Let there be smoke!

-
Image
Last time I introduced my new solar power project.  By the end of that post we were monitoring the power flow from the solar panel to the battery in a simple test environment.  This time we're going to dive in and start trying to control the charge power. What's MPPT? Maximum Power Point Tracking (or Transfer , depending on whom you ask) is a condition where we can transfer the highest power from the solar panel into the battery.  The key here is power, rather than voltage or current on their own.  Remember that power is the product of voltage and current, so it is possible that you will find that perhaps higher current and lower voltage (or vice versa) is the most efficient. What makes this a little problematic is that the peak power point varies constantly depending on the battery state of charge and temperature, so it is not a set and forget operation. This is why the first thing we did was to build a power monitor.  Now we can determine at any point of time...
Read more