带有 CUDA 和 CUDNN 的 DLIB：导入 dlib 错误

Question

我按照在线说明（https://gist.github.com/gipinze/52531983fb47621d51c8d33dacefca49）创建了一个支持 dlib 的 cuda 的 Python venv。 我正在使用：VS 17 2022、cuda 12.5、cudnn 8.9.5 我做到了：

conda create -n dlib-gpu python=3.8 cmake ipython
conda activate dlib-gpu
git clone https://github.com/davisking/dlib.git
cd dlib
mkdir build
cd build
cmake -G "Visual Studio 17 2022" -T host=x64 -DDLIB_USE_CUDA=1 -DUSE_AVX_INSTRUCTIONS=1 ..
cmake --build . --config Release
cd ..
python setup.py install -G "Visual Studio 17 2022" --set USE_AVX_INSTRUCTIONS=1 --set DLIB_USE_CUDA=1 --no DLIB_GIF_SUPPORT

运行所有这些命令后，我尝试在 ipython 中导入 dlib 并收到此错误：

ImportError                               Traceback (most recent call last)
Cell In[1], line 1
----> 1 import dlib

File D:\anaconda3\envs\dlib-gpu\lib\site-packages\dlib-19.24.99-py3.8-win-amd64.egg\dlib\__init__.py:19
     16     add_lib_to_dll_path('D:/cudnn-windows-x86_64-8.9.5.29_cuda12-archive/lib/x64/cudnn.lib')
     17     add_lib_to_dll_path('C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.5/lib/x64/cudart.lib')
---> 19 from _dlib_pybind11 import *
     20 from _dlib_pybind11 import __version__, __time_compiled__

ImportError: DLL load failed while importing _dlib_pybind11: The specified module could not be found.

我尝试过：

pip freeze | grep dlib
dlib==19.24.99

你能帮助我，让我可以在启用 cuda 的情况下使用 dlib 吗？我希望上面链接中的指令为 dlib.DLIB_USE_CUDA 返回 True。非常感谢。

Answer 1

我不知道怎么做，但在 init.py 的“path o naconda3 nvs\dlib-gpu\Lib\site-packages\dlib-19.24.99-py3.11-win-amd64.egg\dlib ”，我将 if '1' == 'ON' 更改为 if 'ON' == 'ON' 并且它有效。我现在可以在我的 GPU 上运行 dlib 了。 在此输入图片描述