安装具有gpu支持的dlib

时间:2020-09-18 05:46:44

标签: dlib cudnn

成功安装后,它还会标识cuda和cudnn路径,如下所示,但是当我将import dlib.cuda作为cuda运行时,仍然出现错误:没有名为'dlib.cuda的模块。

我使用了命令-cmake .. -DDLIB_USE_CUDA = 1 -DUSE_AVX_INSTRUCTIONS = 1

控制台输出如下:

-- Building for: Visual Studio 16 2019
-- Selecting Windows SDK version 10.0.18362.0 to target Windows 10.0.19041.
-- The C compiler identification is MSVC 19.26.28806.0
-- The CXX compiler identification is MSVC 19.26.28806.0
-- Check for working C compiler: C:/Program Files (x86)/Microsoft Visual 
   Studio/2019/Community/VC/Tools/MSVC/14.26.28801/bin/Hostx64/x64/cl.exe
-- Check for working C compiler: C:/Program Files (x86)/Microsoft Visual 
   Studio/2019/Community/VC/Tools/MSVC/14.26.28801/bin/Hostx64/x64/cl.exe -- works
-- Detecting C compiler ABI info
-- Detecting C compiler ABI info - done
-- Detecting C compile features
-- Detecting C compile features - done
-- Check for working CXX compiler: C:/Program Files (x86)/Microsoft Visual 
   Studio/2019/Community/VC/Tools/MSVC/14.26.28801/bin/Hostx64/x64/cl.exe
-- Check for working CXX compiler: C:/Program Files (x86)/Microsoft Visual `enter code 
   here`Studio/2019/Community/VC/Tools/MSVC/14.26.28801/bin/Hostx64/x64/cl.exe -- works
-- Detecting CXX compiler ABI info
-- Detecting CXX compiler ABI info - done
-- Detecting CXX compile features
-- Detecting CXX compile features - done
-- Using CMake version: 3.14.1
-- Compiling dlib version: 19.21.99
-- Looking for sys/types.h
-- Looking for sys/types.h - found
-- Looking for stdint.h
-- Looking for stdint.h - found
-- Looking for stddef.h
-- Looking for stddef.h - found
-- Check size of void*
-- Check size of void* - done
-- Enabling AVX instructions
-- Found system copy of libpng: 
   C:/ProgramData/Anaconda3/Library/lib/libpng.lib;C:/ProgramData/Anaconda3/Library/lib/z.lib
-- Found system copy of libjpeg: C:/ProgramData/Anaconda3/Library/lib/jpeg.lib
-- Searching for BLAS and LAPACK
-- Searching for BLAS and LAPACK
-- Looking for pthread.h
-- Looking for pthread.h - not found
-- Found Threads: TRUE
-- Found CUDA: C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v10.1 (found suitable version 
   "10.1", minimum required is "7.5")
-- Looking for cuDNN install...
-- Found cuDNN: C:/ProgramData/Anaconda3/Library/lib/x64/cudnn.lib
-- Building a CUDA test project to see if your compiler is compatible with CUDA...
-- Building a cuDNN test project to check if you have the right version of cuDNN installed...
-- Enabling CUDA support for dlib.  DLIB WILL USE CUDA
-- C++11 activated.
-- Configuring done
-- Generating done
-- Build files have been written to: C:/Users/Shivalik Singh/Desktop/building dataset/dlib/build

0 个答案:

没有答案