Last active
June 1, 2018 15:43
-
-
Save mikaelhg/8acccad1a40545a70bdeef4709a7c7b5 to your computer and use it in GitHub Desktop.
Building dlib with CUDA and cuDNN
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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