【发布时间】:2019-07-04 02:09:52
【问题描述】:
我正在尝试让 dlib 在 Ubuntu 上使用 CUDA。我知道硬件是有能力的。我已经有了 CUDA 和 cuDNN。
我从 git repo 获得 dlib 并运行以下命令:
cd build
cmake .. -DCMAKE_PREFIX_PATH=/usr/local/cuda
cmake . --build
构建过程说dlib找到了cuda和cuDNN,dlib会使用cuda,但是当我运行python shell时,输入
import dlib
dlib.DLIB_USE_CUDA
它返回 False。
我不确定为什么会发生这种情况。顺便说一句,我正在使用python3。我也已经通过 pip3 安装了 dlib,不知道是不是用错了。
【问题讨论】:
-
您的问题似乎与this one重复
标签: python-3.x ubuntu dlib cudnn