如何安装dlib gpu版本
Posted AI浩
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了如何安装dlib gpu版本相关的知识,希望对你有一定的参考价值。
安装dlib cpu
安装dlib的时候,如果直接使用的是pip,则是不带有GPU版本,在调用执行的时候,使用的是CPU
pip install dlib
安装dlib gpu
如果要使用GPU的话,则需要自己编译dlib,然后再安装的时候设置使用CUDA
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 --set USE_AVX_INSTRUCTIONS=1 --set DLIB_USE_CUDA=1
以上是关于如何安装dlib gpu版本的主要内容,如果未能解决你的问题,请参考以下文章