작성
·
209
0
안녕하세요? 선생님께서 해 주시는 컴퓨터비전 강의를 수강하고 있는데 오류가 계속 발생해서 글을 써 봅니다. 제 데스크탑 내에 GPU가 있어서 데스크탑으로 코드를 수행하고 있었는데 yolo코드를 Tensorflow 1.x 버전에서 수행하려고 하는데 코드가 돌아가지가 않습니다. 제 경우는 일단 선생님께서 알려주신 대로 아나콘다 가상 환경을 만들고 tf113환경에서 코드를 돌리려고 하니
위의 그림과 같이 모듈을 가져올 수 없다는 오류가 계속 뜹니다. 전에 해 주신 Faster-R-CNN, SSD 같은 경우에는 Tensorflow 2.0 환경에서도 몇 가지 함수들만 바꿔서 실행을 해 보니 2.x 환경에서도 코드가 잘 수행이 되었는데 yolo부터는 아예 2.x 환경에서는
이런 식으로 2.x 버전에서 지원하지 않은 함수라고 해서 yolo부분에서부터 선생님께서 알려주신 대로 tf113환경에서 돌리려고 하니 ImportError가 나서 어떻게 하는게 좋을지 선생님께 여쭤보고 싶어서 이렇게 글을 쓰게 되었습니다. 어떻게 하는게 좋을까요?
답변 1
0
안녕하십니까,
오류로만 봐서는 tensorflow를 데스크탑 GPU에서 로딩을 하지 못하는 것으로 보입니다.
import tensorflow가 안되는 것 같습니다.
보시면 DLL에서 Tensorflow를 Loading하지 못하고 있습니다. tensorflow 1.x 버전이 제대로 깔리지 않았거나, GPU CUDA/CUDNN에서 Tensorflow 1.x와 잘 호환이 안되는것 같습니다.
데스크탑에서 tensorflow 2.x는 제대로 구동되나요?
keras yolo의 경우는 소스코드에 Session관련한 코드가 있는데 이게 tensorflow 2.x에서는 동작하지 않기 때문에 오류가 발생합니다.
윈도우에서는 GPU와 Tensorflow, Object Detection 궁합이 잘 맞지 않습니다. 윈도우에서 하시려면 이런 부분들때문에 이슈가 발생하기 쉽습니다. 가급적이면 linux환경이나 colab환경에서 실습을 부탁드립니다.