win11安装dlib并使用cuda加速

1、首先在本地安装好anaconda;
2、若不适用gpu加速的dlib,直接在anaconda的虚拟环境中,使用pip install dlib即可;
3、若想使用gpu加速的dlib,例如调用dlib中的神经网络模型;首先:
检查本地是否安装cuda和cudnn,若没有安装,则先安装他们; 安装CMake,CMake官网下载地址:https://cmake.org/download/,进入下载最新的msi文件,我下载的是:cmake-3.26.0-rc5-windows-x86_64.msi;下载完成双击安装,一路安装即可,只需要注意勾选上一个地方:Add CMake to the system PATH for all users。否则你需要自己在本地配置该环境变量。完成后,直接在本地的IDE或者cmd命令行中,测试一下,输入cmake,会返回一些信息表示安装成功,(有时需要重启一下)
去dlib官网:http://dlib.net/下载最新的dlib压缩包,然后解压,将解压后的文件放入anaconda的虚拟环境中,我放的路径是D:\anaconda\envs\all_test\Lib\site-packages\,然后在pycharm中或者其他方式进入dlib文件夹下,运行:

python setup.py install --no DLIB_GIF_SUPPORT

(命令后跟–no DLIB_GIF_SUPPORT是因为:不跟容易报libfig库相关的错)进行编译安装,在安装的过程中会出现在: Enabling CUDA support for dlib. DLIB WILL USE CUDA,表示dlib可以使用gpu加速。
最后测试:

import dlib
print(dlib.DLIB_USE_CUDA) 
print(dlib.cuda.get_num_devices()) 返回true表示成功
  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值