Cài đặt TensorFlow cho C

Xem trên TensorFlow.org Chạy trong Google Colab Xem nguồn trên GitHub Tải xuống sổ ghi chép

TensorFlow cung cấp một API C có thể được sử dụng để tạo liên kết cho các ngôn ngữ khác . API được định nghĩa trong c_api.h và được thiết kế để đơn giản và đồng nhất hơn là thuận tiện.

Gói Libtensorflow C hàng đêm

Các gói Libtensorflow được xây dựng hàng đêm và tải lên GCS cho tất cả các nền tảng được hỗ trợ. Chúng được tải lên nhóm GCS libtensorflow hàng đêm và được lập chỉ mục theo hệ điều hành và ngày xây dựng. Đối với các đối tượng dùng chung MacOS và Linux, chúng tôi có một tập lệnh đổi tên tệp .so được phiên bản thành ngày hiện tại được sao chép vào thư mục có tạo tác.

Nền tảng được hỗ trợ

TensorFlow cho C được hỗ trợ trên các hệ thống sau:

  • Linux, 64-bit, x86
  • macOS, Phiên bản 10.12.6 (Sierra) trở lên
  • Windows, 64-bit x86

Thành lập

Tải xuống và giải nén

Thư viện TensorFlow C URL
Linux
Chỉ CPU Linux https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.7.0.tar.gz
Hỗ trợ GPU Linux https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-2.7.0.tar.gz
hệ điều hành Mac
chỉ CPU macOS https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-darwin-x86_64-2.7.0.tar.gz
các cửa sổ
Chỉ CPU Windows https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-windows-x86_64-2.7.0.zip
Chỉ GPU Windows https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-windows-x86_64-2.7.0.zip

Giải nén kho lưu trữ đã tải xuống, chứa các tệp tiêu đề để đưa vào chương trình C của bạn và các thư viện được chia sẻ để liên kết với.

Trên Linux và macOS, bạn có thể muốn giải nén sang /usr/local/lib :

FILENAME=libtensorflow-cpu-linux-x86_64-2.7.0.tar.gz
wget -q --no-check-certificate https://storage.googleapis.com/tensorflow/libtensorflow/${FILENAME}
sudo tar -C /usr/local -xzf ${FILENAME}

Người liên kết

Trên Linux / macOS, nếu bạn giải nén thư viện TensorFlow C vào thư mục hệ thống, chẳng hạn như /usr/local , hãy định cấu hình trình liên kết bằng ldconfig :

sudo ldconfig /usr/local/lib

Nếu bạn giải nén thư viện TensorFlow C vào một thư mục không thuộc hệ thống, chẳng hạn như ~/mydir , thì hãy định cấu hình các biến môi trường của trình liên kết:

Linux

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

hệ điều hành Mac

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

Xây dựng

Chương trình ví dụ

Với thư viện TensorFlow C được cài đặt, hãy tạo một chương trình mẫu với mã nguồn sau ( hello_tf.c ):

%%writefile 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;
}
Writing hello_tf.c

Biên dịch

Biên dịch chương trình mẫu để tạo tệp thực thi, sau đó chạy:

gcc hello_tf.c -ltensorflow -o hello_tf

./hello_tf
Hello from TensorFlow C library version 2.7.0-dev20211101

Nếu chương trình không được tạo, hãy đảm bảo rằng gcc có thể truy cập thư viện TensorFlow C. Nếu được giải nén thành /usr/local , hãy chuyển rõ ràng vị trí thư viện cho trình biên dịch:

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

./hello_tf
Hello from TensorFlow C library version 2.7.0-dev20211101

Xây dựng từ nguồn

TensorFlow là mã nguồn mở. Đọc hướng dẫn để xây dựng thư viện C của TensorFlow từ mã nguồn.