当我使用 tensorflow_gpu 时,Dlib 不使用 GPU

Dlib don't using GPU when I using tensorflow_gpu

我有 2 个版本 的 tensorflow、cpu 和 gpu。我正在使用 conda 来管理我的环境。当我激活 cpu 时,dlib 可以在 GPU 上很好地 运行。但是当我使用 gpu 时,它只是 cpu 上的 运行。我不知道为什么,请帮助我。

有一些我的系统信息,不知道能不能给点信息。

测试代码

conda activate ML

ipython

>>> import dlib

>>> dlib.DLIB_USE_CUDA

True

上面的代码输出 True 但下面的代码输出 False。

conda activate ML_GPU

ipython

>>> import dlib

>>> dlib.DLIB_USE_CUDA

False

我希望当我使用 GPU 版本的 tensorflow dlib 时,可以像 CPU 那样使用我的 GPU。有人可以帮忙吗?

我已经通过手动安装 dlib 解决了这个问题。 有一些步骤可以做到这一点。

注意:您应该在虚拟环境中执行这些步骤。

conda activate ML_GPU

pip uninstall dlib

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

最后一步还是有问题。如果您看到这样的信息,您应该注意——已找到 CUDA,但您的编译器未能编译一个简单的 CUDA 程序,因此 dlib 不会使用 CUDA。

那么你应该改变你的g++编译器版本。你也可以从
找一些资料

我的g++版本是:(Ubuntu 6.5.0-2ubuntu1~18.04) 6.5.0 20181026,最新版本似乎与CUDA不兼容。

如果一切正常,你会看到

>>> dlib.DLIB_USE_CUDA
True