小编rah*_*ddy的帖子

如何检查 dlib 是否使用 GPU?

我的机器有 Geforce 940mx GDDR5 GPU。

我已经安装了运行 GPU 加速 dlib(具有 GPU 支持)的所有要求:

  1. CUDA 9.0 工具包,包含来自https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal 的所有 3 个补丁更新

  2. cuDNN 7.1.4

然后我在 Github 上克隆 dlib/davisKing 存储库以编译具有 GPU 支持的 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 --yes USE_AVX_INSTRUCTIONS --yes DLIB_USE_CUDA
Run Code Online (Sandbox Code Playgroud)

现在我怎么可能检查/确认 dlib(或其他库依赖于 dlib,如 Adam Geitgey 的 face_recognition)是否在 python shell/Anaconda(jupyter Notebook)中使用 GPU?

face-recognition python-3.x dlib tensorflow

5
推荐指数
2
解决办法
3万
查看次数

标签 统计

dlib ×1

face-recognition ×1

python-3.x ×1

tensorflow ×1