我用pip安装了dlib.我的显卡支持CUDA,但在运行dlib时,它不使用GPU.
我正在研究ubuntu 18.04
Python 3.6.5 (default, Apr 1 2018, 05:46:30)
[GCC 7.3.0] on linux
>>> import dlib
>>> dlib.DLIB_USE_CUDA
False
我还安装了NVidia Cuda Compile驱动程序,但它仍无法正常工作.
nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Nov__3_21:07:56_CDT_2017
Cuda compilation tools, release 9.1, V9.1.85
任何人都可以帮助我如何让它工作. ？
解决方法:
我有类似的问题,在我的情况下,我错过了cuDNN库,它阻止了dlib使用CUDA指令进行编译,尽管我安装了CUDA编译器和其他驱动程序.
下一部分是从this repo下载dlib.
然后运行此命令以使用CUDA和AVX指令安装dlib,您无需使用make文件使用CMake手动编译它：
python setup.py install --yes USE_AVX_INSTRUCTIONS --yes DLIB_USE_CUDA
现在重要的部分是读取日志,如果python实际上可以找到CUDA,cuDNN并且可以使用CUDA编译器来编译测试项目.这些是重要的路线：
-- Found CUDA: /usr/local/cuda/bin/ (found suitable version "8.0", minimum required is "7.5")
-- Looking for cuDNN install...
-- Found cuDNN: /usr/local/cuda/lib64/libcudnn.so
-- Building a CUDA test project to see if your compiler is compatible with CUDA...
我遇到的第二个问题与CMake版本有关.最新版本有一些已知的cuda和dlib问题,所以我必须安装CMake 3.12.3才能使它工作.
