我使用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编译驱动程序,但它仍然不能工作。
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
有谁能帮我把它弄好吗??
发布于 2018-11-01 17:44:09
我也遇到过类似的问题,虽然我安装了cuDNN编译器和其他驱动程序,但我没有安装CUDA库,它阻止了dlib使用CUDA指令进行编译。
下一部分是从this repo下载dlib。
然后运行这个命令来安装带有CUDA和AVX指令的dlib,你不需要使用make file通过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才能正常工作。
发布于 2019-06-19 00:52:17
与在Windows上一样,有两个不同的问题导致了这一问题:
没有安装installation.
,
如果前两次更正不起作用,请尝试第三个。当时我的CUDA版本是10.1。
发布于 2018-09-20 23:08:22
我们遇到了完全相同的问题,即CUDA驱动程序安装正确,但dlib.DLIB_USE_CUDA标志为'False‘。
通过'pip3 install -v dlib‘安装dlib表明它使用的是不兼容的不同版本的C++编译器。
安装Visual Studio 14 2015为我们解决了这个问题。
需要注意的一件事是,当我们尝试使用源代码中的命令'python setup.py install‘进行安装时,我们收到了dlib将使用cuda的消息,但dlib.DLIB_USE_CUDA标志仍然设置为False。
https://stackoverflow.com/questions/51943050
复制相似问题