作者:手机用户2502872795 | 来源:互联网 | 2023-10-10 11:05
我使用以下命令安装Dlib:
$ workon # optional
$ 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
一切都安装完成后,我尝试运行以下代码:
print(cuda.get_num_devices())
v = dlib.DLIB_USE_CUDA
打印结果给了我1
但v
是False
任何想法可能是什么问题?
我正在使用此图形卡:
Intel(R) UHD Graphics 630
NVIDIA GeForce GTX 1050 Ti
编辑
我现在在编译中发现了这一点:
-- Found Threads: TRUE
CUDA_TOOLKIT_ROOT_DIR not found or specified
-- Could NOT find CUDA (missing: CUDA_TOOLKIT_ROOT_DIR CUDA_NVCC_EXECUTABLE CUDA_INCLUDE_DIRS CUDA_CUDART_libraRY) (Required is at least version "7.5")
-- DID NOT FIND CUDA
-- Disabling CUDA support for dlib. DLIB WILL NOT USE CUDA
任何想法都缺少,需要安装吗?我可以看到所需的CUDA
是7.5
,但是从Nvidia
文档中我的卡支持6.1
,所以我将不能在cuda中使用dlib?
您需要安装 CUDA 工具包和 Cudnn
https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#install-windows