Tensorflow Archives

텐서플로우 C 예제
C언어에서 텐서플로우 API를 사용하는 방법에 대해서 설명드리도록 하겠습니다. 텐서플로우 라이브러리인 libtensorflow 다운로드 후 설치 방법과 예제 소스코드를 사용하여 빌드 후 실행하는 방법에 대해서 함께 설명드리도록 하겠습니다.

텐서플로우 c언어 API는 현재 아래의 플랫폼에서 지원되고 있습니다. 윈도우와 리눅스는 CPU 와 GPU버전을 모두 지원하고 있지만, macOS 환경에서는 CPU 버전만 지원되고 있습니다.

C언어에서 텐서플로우의 API를 사용하기 위해서는 먼저 헤더 파일과 라이브러리등을 설치해 주셔야 합니다. 최신 버전의 텐서플로우 라이브러리 파일인 libtensorflow는 아래 링크에서 다운 받으 실 수 있습니다.

Linux CPU only	https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-1.14.0.tar.gz
Linux GPU support	https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-1.14.0.tar.gz

위의 링크는 Linux 버전의 libtensorflow 파일입니다. 다른 버전의 라이브러리 파일을 다운 받기 위해서는 아래의 페이지를 참고해 주시기 바랍니다.

Linux 또는 macOS 환경일 경우 다운로드 받은 파일을 아래와 같이 /usr/local 디렉터리에 압축 해제 합니다.

$ sudo tar -C /usr/local -xzf libtensorflow-cpu-linux-x86_64-1.14.0.tar.gz

C언어 헤더파일과 라이브러리 파일들이 압축해제 되게 됩니다.

Linux 또는 macOS 환경에서 /usr/local 와 같은 시스템 디렉터리에 텐서플로우 라이브러리 파일을 압축해제 했을 경우 linker 를 설정하기 위해서 다음과 같은 명령어를 실행해 주시기 바랍니다.

$ sudo ldconfig

만약 ~/mydir 와 같은 시스템 디렉터리가 아닌 위치에 libtensorflow 파일을 해제 했을 경우 아래와 같이 환경 변수를 설정해 주시기 바랍니다.

export LIBRARY_PATH=$LIBRARY_PATH:~/mydir/lib
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:~/mydir/lib

export LIBRARY_PATH=$LIBRARY_PATH:~/mydir/lib
export DYLD_LIBRARY_PATH=$DYLD_LIBRARY_PATH:~/mydir/lib

다음과 같은 예제 소스코드를 생성하여, 빌드 후 실행해 보도록 하겠습니다. 파일이름은 hello.c 이라고 설정하도도록 하겠습니다.

#include <stdio.h>
#include <tensorflow/c/c_api.h>

int main() {
  printf("TensorFlow C library version %s\n", TF_Version());
  return 0;
}

다음 명령어를 실행하여 소스코드를 빌드합니다.

$ gcc hello.c -ltensorflow -o hello

만약 위의 명령어로 소스코드가 빌드가 되지 않는 다으면, 다음과 같이 헤더 파일과 라이브러리 디렉터리를 명시적으로 지정하면서 빌드 하실 수도 있습니다.

$ gcc -I/usr/local/include -L/usr/local/lib hello.c -ltensorflow -o hello

빌드가 성공적으로 이뤄지게 되면 hello 파일이 생성됩니다. 생성된 hello 파일을 실행하면 다음과 같이 Tensorflow 버전이 출력되는 것을 확인 할 수 있습니다.

텐서플로우 C API와 관련된 보다 자세한 내용은 아래의 페이지를 참고해 주시기 바랍니다.

( 본문 인용시 출처를 밝혀 주시면 감사하겠습니다.)

HiSEON