'caffe'에 해당되는 글 2건

터미널에서 가상환경을 activate 하고 import caffe 를 하면 잘 동작하지만 Pycharm 에서 ssh interpreter 로 같은 가상환경을 사용해 import caffe 를 할 경우 caffe 를 찾을 수 없다고 에러가 나는 경우가 있다

가상환경 패키지 차이겠지만 virtualenv 의 경우 환경변수를 직접 추가해줘야한다

두 가지 방법이 있는데

첫 번재 방법은 코드에다가 환경변수를 직접 추가하는 방법

import sys
sys.path.append("/home/ubuntu/caffe/python")

두 번째 방법은 virtualenv 패키지에 환경변수를 추가해주는 방법이다

$ add2virtualenv /home/ubuntu/caffe/python

caffe 를 빌드하다보면 참 많은 에러를 만나게 되는데 그 중 가장 잔인한 에러는 build가 끝나고 runtest 할 때 발생하는 에러라고 생각한다...(희망고문)

.build_release/tools/caffe
.build_release/tools/caffe: error while loading shared libraries: libcudart.so.9.0: cannot open shared object file: No such file or directory
Makefile:533: recipe for target 'runtest' failed
make: *** [runtest] Error 127

이 에러가 발생했다면 우선 ~/.bashrc 에 환경변수를 추가해준다

bashrc 에 위의 두 줄을 추가해준 뒤 터미널로 나와 마지막 명령어를 실행한다.

export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda-9.0/bin:$PATH


source ~/.bashrc

이 방법을 사용하고나서도 runtest 에러가 발생한다면 아래의 명령어를 통해 Library 의 링크를 직접 추가해줘야 한다.

** 먼저 해당 경로에 들어가 파일이 존재하는지 확인부터 하는게 좋다

sudo cp /usr/local/cuda-9.0/lib64/libcudart.so.9.0 /usr/local/lib/libcudart.so.9.0 && sudo ldconfig
sudo cp /usr/local/cuda-9.0/lib64/libcublas.so.9.0 /usr/local/lib/libcublas.so.9.0 && sudo ldconfig
sudo cp /usr/local/cuda-9.0/lib64/libcurand.so.9.0 /usr/local/lib/libcurand.so.9.0 && sudo ldconfig
sudo cp /usr/local/cuda-9.0/lib64/libcudnn.so.7 /usr/local/lib/libcudnn.so.7 && sudo ldconfig

runtnest 가 성공하면 이런 화면이 나와야 한다