當前位置: 首頁>>技術教程>>正文


安裝TensorFlow C語言版

TensorFlow在文件c_api.h中定義的C語言API,可用於構建其他語言的綁定

支持的平台

您可以在以下操作係統上安裝TensorFlow for C:

  • Linux
  • Mac OS X
  • 安裝

    以下步驟用於安裝TensorFlow for C:

    1. 確定是否需要GPU支持,參閱“確定要安裝的TensorFlow”的部分。

    2. 下載並提取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

    3. 在步驟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中指出的環境變量安裝
    • 是否導出了這些環境變量?

    參考資料

    本文由《純淨天空》出品。文章地址: https://vimsky.com/zh-tw/article/3625.html,未經允許,請勿轉載。