colab에 CUDA OpenCV 설치하기

from google.colab import drive
drive.mount('/content/gdrive')

!git clone https://github.com/opencv/opencv.git

!git clone https://github.com/opencv/opencv_contrib.git

!mkdir /content/build
%cd /content/build

!cmake -DOPENCV_EXTRA_MODULES_PATH=/content/opencv_contrib/modules  \
       -DBUILD_SHARED_LIBS=OFF \
       -DBUILD_TESTS=OFF \
       -DBUILD_PERF_TESTS=OFF \
       -DBUILD_EXAMPLES=OFF \
       -DWITH_OPENEXR=OFF \
       -DWITH_CUDA=ON \
       -DWITH_CUBLAS=ON \
       -DWITH_CUDNN=ON \
       -DOPENCV_DNN_CUDA=ON \
       /content/opencv

 

!make -j8 install

여기서 엄청 걸림.... 한 2~4시간 정도

 

!cp  /content/build/lib/python3/cv2.cpython-36m-x86_64-linux-gnu.so   "/content/gdrive/MyDrive/cv2_cuda"

이제 다운받은 파일을 나중에 쓸일이 있을지도 모르니깐 나의 gdrive에 cv2_cuda에 넣어놓는다.

cv2_cuda는 !mkdir 로 만든 폴더임.

 

 

다음으로 미리 설치된 cv파일을 제거하고, 방금 설치한 cv파일을 넣는다!

!pip uninstall opencv-python
%cd /content/gdrive/MyDrive/cv2_cuda/
!cp cv2.cpython-36m-x86_64-linux-gnu.so /usr/local/lib/python3.7/dist-packages

이 다운받은 파일을 /usr/local/lib/python3.7(현재 내 코랩은 3.7버전으로 진행되고 있었음)/dist-packages로 넣어준다.

기존의 cv파일이 이 dist-packages안에 있었는데 그걸 제거하고 새로운 cv파일을 넣는 것 같다.

 

import cv2

print(cv2.__version__)
print(cv2.cuda.getCudaEnableDeviceCount())

import를 해주고, 버전을 확인한다음 cuda가 가능한지 물어본다. 1이 나오면 성공, 0이 나오면 어디선가 오류가 생긴 거다.

 

 

 

아래 두페이지에서 참고해서 opencv를 설치했다.

 

https://answers.opencv.org/question/233476/how-to-make-opencv-use-gpu-on-google-colab/

how to make openCV use GPU on google colab - OpenCV Q&A Forum

https://hanryang1125.tistory.com/18?category=902300 

Google Colaboratory에 CUDA - OpenCV 설치하기 (Colab)