我使用 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 编译器和其他驱动程序。
下一部分是从下载 dlib这个仓库 https://github.com/davisking/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 才能使其正常工作。
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
