CV-CUDA is an open-source project that enables building efficient cloud-scale Artificial Intelligence (AI) imaging and computer vision (CV) applications. It uses graphics processing unit (GPU) acceleration to help developers build highly efficient pre- and post-processing pipelines. CV-CUDA originated as a collaborative effort between NVIDIA and ByteDance.
Refer to our Developer Guide for more information on the operators available as of release v0.2.1-alpha.
To get a local copy up and running follow these steps.
- Linux distro:
- Ubuntu x86_64 >= 18.04
- WSL2 with Ubuntu >= 20.04 (tested with 20.04)
- NVIDIA driver
- Linux: Driver version 520.56.06 or higher
- CUDA Toolkit
- Version 11.7 or above. (12.0 is not yet tested.)
- GCC >= 11.0
- Python >= 3.7
- cmake >= 3.22
The following steps describe how to install CV-CUDA from pre-built install packages. Choose the installation method that meets your environment needs.
tar -xvf nvcv-lib-0.2.1-cuda11-x86_64-linux.tar.xz
tar -xvf nvcv-dev-0.2.1-cuda11-x86_64-linux.tar.xz
sudo apt-get install -y ./nvcv-lib-0.2.1-cuda11-x86_64-linux.deb ./nvcv-dev-0.2.1-cuda11-x86_64-linux.deb
pip install nvcv_python-0.2.1-cp38-cp38-linux_x86_64.whl
Follow these instruction to build CV-CUDA from source:
-
Set up your local CV-CUDA repository
-
Install prerequisites needed to setup up the repository.
On Ubuntu 22.04, install the following packages:
- git-lfs: to retrieve binary files from remote repository
sudo apt-get install -y git git-lfs
-
After cloning the repository (assuming it was cloned in
~/cvcuda
), it needs to be properly configured by running theinit_repo.sh
script only once.cd ~/cvcuda ./init_repo.sh
-
-
Build CV-CUDA
-
Install the dependencies required for building CV-CUDA
On Ubuntu 22.04, install the following packages:
- g++-11: compiler to be used
- cmake, ninja-build (optional): manage build rules
- python3-dev: for python bindings
- libssl-dev: needed by the testsuite (MD5 hashing utilities)
sudo apt-get install -y g++-11 cmake ninja-build python3-dev libssl-dev
For CUDA Toolkit, any version of the 11.x series should work. CV-CUDA was tested with 11.7, thus it should be preferred.
sudo apt-get install -y cuda-minimal-build-11-7
-
Build the project
ci/build.sh
This will compile a x86 release build of CV-CUDA inside
build-rel
directory. The library is in build-rel/lib, docs in build-rel/docs and executables (tests, etc...) are in build-rel/bin.The script accepts some parameters to control the creation of the build tree:
ci/build.sh [release|debug] [output build tree path]
By default it builds for release.
If output build tree path isn't specified, it'll be
build-rel
for release builds, andbuild-deb
for debug.
-
-
Build Documentation
-
Install the dependencies required for building the documentation
On Ubuntu 22.04, install the following packages:
- doxygen: parse header files for reference documentation
- python3, python3-pip: to install some python packages needed
- sphinx, breathe, exhale, recommonmark, graphiviz: to render the documentation
- sphinx-rtd-theme: documenation theme used
sudo apt-get install -y doxygen graphviz python3 python3-pip sudo python3 -m pip install sphinx==4.5.0 breathe exhale recommonmark graphviz sphinx-rtd-theme
-
Build the documentation
ci/build_docs.sh [build folder]
Example:
ci/build_docs.sh build_docs
-
-
Build and run Samples
- For instructions on how to build samples from source and run them, see the Samples documentation.
-
Run Tests
-
Install the dependencies required for running the tests
On Ubuntu 22.04, install the following packages:
- python3, python3-pip: to run python bindings tests
- torch: dependencies needed by python bindings tests
sudo apt-get install -y python3 python3-pip sudo python3 -m pip install pytest torch
-
Run the tests
The tests are in
<buildtree>/bin
. You can run the script below to run all tests at once. Here's an example when build tree is created inbuild-rel
build-rel/bin/run_tests.sh
-
-
Package installers
Installers can be generated using the following cpack command once you have successfully built the project
cd build-rel cpack .
This will generate in the build directory both Debian installers and tarballs (*.tar.xz), needed for integration in other distros.
For a fine-grained choice of what installers to generate, the full syntax is:
cpack . -G [DEB|TXZ]
- DEB for Debian packages
- TXZ for *.tar.xz tarballs.
CV-CUDA is an open source project. As part of the Open Source Community, we are committed to the cycle of learning, improving, and updating that makes this community thrive. However, as of release v0.2.1-alpha, CV-CUDA is not yet ready for external contributions.
To understand the process for contributing the CV-CUDA, see our Contributing page. To understand our committment to the Open Source Community, and providing an environment that both supports and respects the efforts of all contributors, please read our Code of Conduct.
CV-CUDA operates under the Apache-2.0 license.
CV-CUDA, as a NVIDIA program, is committed to secure development practices. Please read our Security page to learn more.
CV-CUDA is developed jointly by NVIDIA and ByteDance.