Google cam kết thúc đẩy bình đẳng chủng tộc đối với cộng đồng đen. Xem cách.
Trang này được dịch bởi Cloud Translation API.
Switch to English

Cài đặt TensorFlow cho Go

TensorFlow cung cấp một API Go - đặc biệt hữu ích cho tải mô hình tạo ra với Python và chạy chúng trong một ứng dụng Go.

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

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

  • Linux, 64-bit, x86
  • hệ điều hành MacOS, phiên bản 10.12.6 (Sierra) hoặc cao hơn

Thiết lập

thư viện TensorFlow C

Cài đặt thư viện TensorFlow C đó là cần thiết cho gói TensorFlow Go.

Tải xuống

Tải về và cài đặt gói TensorFlow Go và phụ thuộc của nó:

go get github.com/tensorflow/tensorflow/tensorflow/go

Và xác nhận cài đặt của bạn:

go test github.com/tensorflow/tensorflow/tensorflow/go

Xây dựng

chương trình ví dụ

Với gói TensorFlow Go cài đặt, tạo ra một chương trình ví dụ với mã nguồn sau ( hello_tf.go ):

 package main

import (
    tf "github.com/tensorflow/tensorflow/tensorflow/go"
    "github.com/tensorflow/tensorflow/tensorflow/go/op"
    "fmt"
)

func main() {
    // Construct a graph with an operation that produces a string constant.
    s := op.NewScope()
    c := op.Const(s, "Hello from TensorFlow version " + tf.Version())
    graph, err := s.Finalize()
    if err != nil {
        panic(err)
    }

    // Execute the graph in a session.
    sess, err := tf.NewSession(graph, nil)
    if err != nil {
        panic(err)
    }
    output, err := sess.Run(nil, []tf.Output{c}, nil)
    if err != nil {
        panic(err)
    }
    fmt.Println(output[0].Value())
}
 

Chạy

Chạy chương trình ví dụ:

go run hello_tf.go

Các đầu ra lệnh: Hello from TensorFlow version number

Chương trình này có thể tạo ra các thông điệp cảnh báo sau đây, bạn có thể bỏ qua:

W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library
wasn't compiled to use *Type* instructions, but these are available on your
machine and could speed up CPU computations.

Xây dựng từ nguồn

TensorFlow là mã nguồn mở. Đọc các hướng dẫn để xây dựng TensorFlow cho Go từ mã nguồn.