用cuda编译dlib

Question

我正在尝试用 cuda 编译 dlib。我从此 GitHub 链接https://github.com/davisking/dlib克隆存储库，然后尝试运行

python setup.py install --yes USE_AVX_INSTRUCTIONS

但我收到一个错误。任何人都可以帮我解决它吗？

Answer 1

创建一个文件夹来托管源

mkdir -p /opt/SP/packages/
cd $_

克隆存储库

git clone https://github.com/davisking/dlib.git
cd dlib
git submodule init
git submodule update

创建一个用于构建软件的文件夹（必须安装 cmake）

mkdir build
cd $_

现在使用以下选项运行 cmake：

cmake  -D DLIB_USE_CUDA=1 -D USE_AVX_INSTRUCTIONS=1 ../

如果一切正常，您可以看到类似的输出

注意：如果您的系统与它们兼容，请安装cuDNN、openblas 和Intel MKL/BLAS/LAPACK。

英特尔性能库链接：https : //software.seek.intel.com/performance-libraries

cuDNN 链接：https : //developer.nvidia.com/cudnn

现在你可以编译源代码

cmake --build . --config Release

在这些步骤之后，源代码被编译。

现在您可以安装 python API。

cd ../
python setup.py install

注意：这次编译会使用所有可用的CPU，请确保您有足够的内存进行编译。

完成后，您可以查询您的 python 包以确保 dlib 现在已安装：

点冻结| grep dlib

检查安装是否成功：

打开终端并运行以下命令

>>> import dlib.cuda as cuda;
>>> print(cuda.get_num_devices());
1
>>> import dlib
>>> dlib.DLIB_USE_BLAS
True
>>> dlib.DLIB_USE_CUDA
True
>>> dlib.DLIB_USE_LAPACK
True