0、网上找的步骤如下:
git clone https://github.com/davisking/dlib.git
cd dlib
mkdir build
cd build
cmake .. -DDLIB_USE_CUDA=1 -DUSE_AVX_INSTRUCTIONS=1
cmake --build .
cd ..
python setup.py install --set USE_AVX_INSTRUCTIONS=1 --set DLIB_USE_CUDA=1
按照上面步骤安装的时候遇到两个问题及解决方法如下:
1、无法启用CUDA,报错如下:
CMake Warning at dlib/CMakeLists.txt:614 (message):
You have CUDA installed, but we can't use it unless you put visual studio
in 64bit mode.
Found CUDA, but CMake was unable to find the cuBLAS libraries that should be part of every basic CUDA install. Your CUDA install is somehow broken or incomplete. Since cuBLAS is required for dlib to use CUDA we won't use CUDA.
-- DID NOT FIND CUDA
-- Disabling CUDA support for dlib. DLIB WILL NOT USE CUDA
-- C++11 activated.
-- Configuring done
-- Generating done
-- Build files have been written to: E:/dlib/dlib/build
开始的解决办法:
cmake .. -DDLIB_USE_CUDA=1 -DUSE_AVX_INSTRUCTIONS=1 改为
cmake -A x64 .. -DDLIB_USE_CUDA=1 -DUSE_AVX_INSTRUCTIONS=1
但是到了安装的时候出错,最后参考:
使用cmake-gui生成工程,然后直接安装解决问题。
2、报错--yes has been removed,解决办法:
参考了:https://blog.csdn.net/qq_41375609/article/details/111001589
使dlibgpu用管理员权限打开CMD,
python setup.py install --set USE_AVX_INSTRUCTIONS=1 --set DLIB_USE_CUDA=1改为
python setup.py install --set USE_AVX_INSTRUCTIONS=1 --set DLIB_USE_CUDA=1