./build 실행 시 질문

[shmsw25]

안녕하세요. ./build.sh를 할 때, build.sh의 마지막줄(g++)에서 에러가 나는데요,

File 'string', line 1, in ImportError : No module named 'tensorflow' In file included from ops/count_sketch.cc:5.0: ops/count_sketch.h:4:61: fatal error : third_party/eigen3/unsupported/Eigen/CXX11/Tensor : No such file or directory

라고 합니다.

gcc가 tensorflow 모듈을 인식을 못하는 것 같은데요 제가 tensorflow를 virtual environment에 설치했습니다. which g++이라고 하면 /usr/bin/g++이 뜨고 which python이라고 하면 /home/usrname/venv/bin/python이 뜨고요 g++의 경로와 python의 경로가 다른 것이 원인인가요? 혹시 작성자분은 virtual environment를 사용했는지, 어떻게 하셨는지 궁금합니다

[therne]

g++은 정상인데, virtualenv때문인듯 보입니다. 이 레포지터리를 클론하신 폴더에도 virtual env를 여셔서 텐서플로우를 설치해주셔야 할 것 같아요. 전 virtualenv를 사용하지 않아서 그냥 임포트해도 문제가 없었습니다.

아니면 어차피 native op 빌드 과정에서 텐서플로우를 사용하는 이유는 단순히 헤더 폴더 경로를 알아오기 위해서기때문에 그냥 import tensorflow as tf; print(tf.sysconfig.get_include()) 내용을 따로 복사하셔서 build.sh의 첫번째 TF_INC에 그대로 넣어주셔도 될 것 같아요

[shmsw25]

감사합니다. 말씀하신 대로 하여 해결했습니다.

도움이 정말 많이 되었네요. 올려주신 CBP 덕분에 MCB for Visual QA를 구현했는데 혹시 나중에 논문 전체를 구현하게 된다면 참고해보세요. https://github.com/shmsw25/mcb-model-for-vqa