A
현재 배포된 KSB 툴박스 1.0 버전은 tensorflow CPU 버전을 탑재하고 있으며, CPU를 이용해 딥러닝 학습하는 예제가 탑재되어 있습니다.
Tensorflow GPU를 이용하며 KSB 프레임워크와 연동하는 방법은 2가지가 있습니다.
첫째, tensorflow GPU 버전을 탑재한 KSB 툴박스 1.0 버전은 제한된 환경의 도커이미지만을 제공합니다.
현재 Titan X, 1080 GPU 보드에서 테스트된 CUDA 9.0, CUDNN 7.3.1 버전을 탑재한 docker 이미지와 메뉴얼을 제공하고 있습니다.
사용자의 다양한 GPU 환경과 tensorflow 버전, CUDA 버전, CUDNN 버전을 모두 지원하기에는 무리가 있는 점 양해 부탁드립니다.
참고 링크 : https://csleoss.etri.re.kr/images/contents/manual_1.0/1.2.HowToInstallKsbToolBox-nvidia.html
둘째, tensorflow GPU를 이용하는 다른 방법은 KSB 툴박스를 이용하지 않고, 직접 우분투 서버(베어메탈)에 설치하시는 방법이 있습니다.
우선 사용자의 환경에 맞도록 NVIDIA 드라이버와 CUDA, CUDNN, tensorflow 등을 별도로 설치하시고,
KSB 홈페이지에서 제공하는 "KSB 인공지능 프레임워크 설치" 메뉴얼을 참고하셔서 설치하시면 가능합니다.
참고 링크 : https://csleoss.etri.re.kr/images/contents/manual_1.0/1.3.HowToInstallKsbFrameWork.html