dlib GPU 버전 우분투에 설치 중 경험했던 2가지 문제 해결방법

OS/Linux

bskyvision.com 2021. 1. 23. 09:35

dlib을 GPU 버전으로 설치하려면, 다음과 같은 과정을 따라야합니다.

$ cd dlib

$ mkdir build

$ cd build

$ cmake .. -DDLIB_USE_CUDA=1 -DUSE_AVX_INSTRUCTIONS=1

그런데 저는 이때 다음과 같은 에러 메시지가 떴습니다.

Could NOT find CUDA

이때 저는 stackoverflow.com/questions/19980412/how-to-let-cmake-find-cuda를 참고하여 다음과 같이 명령을 해주니 문제가 해결되었습니다.

$ cmake -D CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-11.2 ..

저의 경우에는 위 명령을 준 후에 다시

$ cmake .. -DDLIB_USE_CUDA=1 -DUSE_AVX_INSTRUCTIONS=1

를 해주니 됐습니다. 제대로 되었을 경우 다음과 같이 뜹니다.

Enabling CUDA support for dlib. DLIB WILL USE CUDA

그 다음에 이어서

$ cmake --build .

명령을 줬습니다. 그런데 이때 저는 또 다음과 같은 에러가 출력되었습니다.

fatal error: X11/Xlib.h: 그런 파일이나 디렉터리가 없습니다

$ apt search Xlib.h

$ sudo apt install libx11-dev

를 명령해준 후에 다시

$ cmake --build .

입력하니 문제 없이 잘 진행되었습니다.

그 다음에 이어서

$ build ..

$ sudo python3 setup.py install

를 명령해줬는데, 여기서는 문제가 없이 잘 흘러갔습니다.

이렇게 해서 드디어 CUDA 사용이 가능한 dlib 설치가 완료되었습니다.