Skip to content

Instantly share code, notes, and snippets.

@mikaelhg
Last active June 1, 2018 15:43
Show Gist options
  • Save mikaelhg/8acccad1a40545a70bdeef4709a7c7b5 to your computer and use it in GitHub Desktop.
Save mikaelhg/8acccad1a40545a70bdeef4709a7c7b5 to your computer and use it in GitHub Desktop.
Building dlib with CUDA and cuDNN
mkdir build; cd build;
cmake .. -DCMAKE_PREFIX_PATH=/usr/local/cuda-9.0 -DCUDA_HOST_COMPILER=/usr/bin/g++-6 -DDLIB_USE_CUDA=1
cmake --build .
python3 setup.py bdist_wheel --set CMAKE_PREFIX_PATH=/usr/local/cuda-9.0 --set CUDA_HOST_COMPILER=/usr/bin/g++-6 --set DLIB_USE_CUDA=1 --set PYTHON_EXECUTABLE=/usr/bin/python3
pip3 install --user dist/dlib-19.13.99-cp36-cp36m-linux_x86_64.whl
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment