TensorFlow在文件c_api.h
中定義的C語言API,可用於構建其他語言的綁定。
支持的平台
您可以在以下操作係統上安裝TensorFlow for C:
安裝
以下步驟用於安裝TensorFlow for C:
-
確定是否需要GPU支持,參閱“確定要安裝的TensorFlow”的部分。
-
下載並提取TensorFlow C庫
/usr/local/lib
, 使用以下shell命令:TF_TYPE="cpu" # Change to "gpu" for GPU support OS="linux" # Change to "darwin" for Mac OS TARGET_DIRECTORY="/usr/local" curl -L \ "https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-${TF_TYPE}-${OS}-x86_64-1.3.0.tar.gz" | sudo tar -C $TARGET_DIRECTORY -xz
tar
命令將TensorFlow C庫提取到lib
的子目錄TARGET_DIRECTORY
。 -
在步驟2中,如果指定了
係統
目錄(例如,/usr/local
)作為TARGET_DIRECTORY
,那麽可以直接運行ldconfig
配置鏈接器。例如:sudo ldconfig
如果指定的不是係統目錄(例如,
~/mydir
),那麽你必須附加提取目錄(例如,~/mydir/lib
)到兩個環境變量。例如:export LIBRARY_PATH=$LIBRARY_PATH:~/mydir/lib # For both Linux and Mac OS X export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:~/mydir/lib # For Linux only export DYLD_LIBRARY_PATH=$DYLD_LIBRARY_PATH:~/mydir/lib # For Mac OS X only
驗證您的安裝
安裝TensorFlow for C後,將以下代碼輸入到一個名為hello_tf.c
的文件中:
#include <stdio.h>
#include <tensorflow/c/c_api.h>
int main() {
printf("Hello from TensorFlow C library version %s\n", TF_Version());
return 0;
}
構建和運行
編譯hello_tf.c
,命令如下:
gcc hello_tf.c
運行生成的可執行文件應該輸出以下消息:
a.out Hello from TensorFlow C library version number
故障排除
如果構建程序失敗,最可能的原因是gcc
找不到TensorFlow C庫。解決這個問題的一個方法是給gcc
指定-I
和-L
選項。例如,如果TARGET_LIBRARY
是/usr/local
目錄,那麽使用gcc
如下:
gcc -I/usr/local/include -L/usr/local/lib hello_tf.c -ltensorflow
如果執行a.out
失敗了,可能是以下問題:
- 程序編譯有沒有錯誤?
- 是否將正確的目錄分配給步驟3中指出的環境變量安裝?
- 是否導出了這些環境變量?