Skip to content
Snippets Groups Projects
Unverified Commit c2ccc5ad authored by William Muir's avatar William Muir Committed by GitHub
Browse files

[Go] Update Go API installation guide for TensorFlow 2.12.0 (#144)

parent 23986ca5
No related branches found
No related tags found
No related merge requests found
Show whitespace changes
Inline Side-by-side
golang_install_guide/README.md
+ 20
− 9
View file @ c2ccc5ad
...@@ -48,9 +48,8 @@ repo, and with one of the following import statements: ...@@ -48,9 +48,8 @@ repo, and with one of the following import statements:
| TensorFlow C API | Graft | | TensorFlow C API | Graft |
| :------------------------ | :-------------------------------------------------------------------------------------------------- | | :------------------------ | :-------------------------------------------------------------------------------------------------- |
| TensorFlow Release 2.11.0 | [`go get github.com/wamuir/graft/tensorflow@v0.3.0`](https://github.com/wamuir/graft/tree/v0.3.0) | | TensorFlow Release 2.12.0 | [`go get github.com/wamuir/graft/tensorflow@v0.4.0`](https://github.com/wamuir/graft/tree/v0.4.0) |
| TensorFlow Release 2.10.1 | [`go get github.com/wamuir/graft/tensorflow@v0.2.1`](https://github.com/wamuir/graft/tree/v0.2.1) | | TensorFlow Release 2.11.1 | [`go get github.com/wamuir/graft/tensorflow@v0.3.1`](https://github.com/wamuir/graft/tree/v0.3.1) |
| TensorFlow Release 2.9.3 | [`go get github.com/wamuir/graft/tensorflow@v0.1.2`](https://github.com/wamuir/graft/tree/v0.1.2) |
| TensorFlow Nightly | [`go get github.com/wamuir/graft/tensorflow@nightly`](https://github.com/wamuir/graft/tree/nightly) | | TensorFlow Nightly | [`go get github.com/wamuir/graft/tensorflow@nightly`](https://github.com/wamuir/graft/tree/nightly) |
...@@ -59,6 +58,8 @@ repo, and with one of the following import statements: ...@@ -59,6 +58,8 @@ repo, and with one of the following import statements:
<details> <details>
<summary>Click to expand</summary> <summary>Click to expand</summary>
> Note: these build instructions are specific to TensorFlow 2.12.0
### 1. Install the TensorFlow C Library ### 1. Install the TensorFlow C Library
Install the [TensorFlow C library](https://www.tensorflow.org/install/lang_c). This Install the [TensorFlow C library](https://www.tensorflow.org/install/lang_c). This
...@@ -66,7 +67,7 @@ library is required for use of the TensorFlow Go package at runtime. For example ...@@ -66,7 +67,7 @@ library is required for use of the TensorFlow Go package at runtime. For example
on Linux (64-bit, x86): on Linux (64-bit, x86):
```sh ```sh
$ curl -L https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.11.0.tar.gz | tar xz --directory /usr/local $ curl -L https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.12.0.tar.gz | tar xz --directory /usr/local
$ ldconfig $ ldconfig
``` ```
...@@ -104,7 +105,7 @@ Instead, follow these instructions.*** ...@@ -104,7 +105,7 @@ Instead, follow these instructions.***
workspace for `/go` in the command below. workspace for `/go` in the command below.
```sh ```sh
$ git clone --branch v2.11.0 https://github.com/tensorflow/tensorflow.git /go/src/github.com/tensorflow/tensorflow $ git clone --branch v2.12.0 https://github.com/tensorflow/tensorflow.git /go/src/github.com/tensorflow/tensorflow
``` ```
- Change the working directory to the base of the cloned TensorFlow repository, - Change the working directory to the base of the cloned TensorFlow repository,
...@@ -121,10 +122,20 @@ Instead, follow these instructions.*** ...@@ -121,10 +122,20 @@ Instead, follow these instructions.***
$ go mod init github.com/tensorflow/tensorflow $ go mod init github.com/tensorflow/tensorflow
``` ```
- Patch tensorflow/go/genop to generate TSL protobufs. - Patch protos to declare Go package.
```sh
$ sed -i '4 i option go_package = "github.com\/tensorflow\/tensorflow\/tensorflow\/go\/core\/framework\/dataset_go_proto";' tensorflow/core/framework/dataset.proto
$ sed -i '4 i option go_package = "github.com\/tensorflow\/tensorflow\/tensorflow\/go\/core\/framework\/optimized_function_graph_go_proto";' tensorflow/core/framework/optimized_function_graph.proto
$ sed -i '4 i option go_package = "github.com\/google\/tsl\/tsl\/go\/core\/protobuf\/for_core_protos_go_proto";;' tensorflow/tsl/protobuf/test_log.proto
```
- Patch tensorflow/go/genop to generate TF and TSL protobufs.
```sh ```sh
$ sed -i '72 i \ ${TF_DIR}\/tensorflow\/tsl\/protobuf\/*.proto \\' tensorflow/go/genop/generate.sh $ sed -i '71d;72d' tensorflow/go/genop/generate.sh
$ sed -i '71 i \ ${TF_DIR}\/tensorflow\/tsl\/protobuf\/*.proto \\' tensorflow/go/genop/generate.sh
$ sed -i '72 i \ ${TF_DIR}\/tensorflow\/compiler\/xla\/stream_executor\/dnn.proto; do' tensorflow/go/genop/generate.sh
``` ```
- Generate wrappers and protocol buffers. - Generate wrappers and protocol buffers.
...@@ -171,7 +182,7 @@ workspace for `/go` in the command below: ...@@ -171,7 +182,7 @@ workspace for `/go` in the command below:
```sh ```sh
$ go mod init hello-world $ go mod init hello-world
$ go mod edit -require github.com/google/tsl@v0.0.0+incompatible $ go mod edit -require github.com/google/tsl@v0.0.0+incompatible
$ go mod edit -require github.com/tensorflow/tensorflow@v2.11.0+incompatible $ go mod edit -require github.com/tensorflow/tensorflow@v2.12.0+incompatible
$ go mod edit -replace github.com/google/tsl=/go/src/github.com/google/tsl $ go mod edit -replace github.com/google/tsl=/go/src/github.com/google/tsl
$ go mod edit -replace github.com/tensorflow/tensorflow=/go/src/github.com/tensorflow/tensorflow $ go mod edit -replace github.com/tensorflow/tensorflow=/go/src/github.com/tensorflow/tensorflow
$ go mod tidy $ go mod tidy
...@@ -219,7 +230,7 @@ func main() { ...@@ -219,7 +230,7 @@ func main() {
```sh ```sh
$ go mod init app $ go mod init app
$ go mod edit -require github.com/google/tsl@v0.0.0+incompatible $ go mod edit -require github.com/google/tsl@v0.0.0+incompatible
$ go mod edit -require github.com/tensorflow/tensorflow@v2.11.0+incompatible $ go mod edit -require github.com/tensorflow/tensorflow@v2.12.0+incompatible
$ go mod edit -replace github.com/google/tsl=/go/src/github.com/google/tsl $ go mod edit -replace github.com/google/tsl=/go/src/github.com/google/tsl
$ go mod edit -replace github.com/tensorflow/tensorflow=/go/src/github.com/tensorflow/tensorflow $ go mod edit -replace github.com/tensorflow/tensorflow=/go/src/github.com/tensorflow/tensorflow
$ go mod tidy $ go mod tidy
......
golang_install_guide/example-program/Dockerfile
+ 11
− 6
View file @ c2ccc5ad
...@@ -14,10 +14,10 @@ ...@@ -14,10 +14,10 @@
# ============================================================================ # ============================================================================
FROM golang:1.19-bullseye FROM golang:1.20-bullseye
# 1. Install the TensorFlow C Library (v2.11.0). # 1. Install the TensorFlow C Library (v2.12.0).
RUN curl -L https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-$(uname -m)-2.11.0.tar.gz \ RUN curl -L https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-$(uname -m)-2.12.0.tar.gz \
| tar xz --directory /usr/local \ | tar xz --directory /usr/local \
&& ldconfig && ldconfig
...@@ -27,10 +27,15 @@ RUN apt-get update && apt-get -y install --no-install-recommends \ ...@@ -27,10 +27,15 @@ RUN apt-get update && apt-get -y install --no-install-recommends \
protobuf-compiler protobuf-compiler
# 3. Install and Setup the TensorFlow Go API. # 3. Install and Setup the TensorFlow Go API.
RUN git clone --branch=v2.11.0 https://github.com/tensorflow/tensorflow.git /go/src/github.com/tensorflow/tensorflow \ RUN git clone --branch=v2.12.0 https://github.com/tensorflow/tensorflow.git /go/src/github.com/tensorflow/tensorflow \
&& cd /go/src/github.com/tensorflow/tensorflow \ && cd /go/src/github.com/tensorflow/tensorflow \
&& go mod init github.com/tensorflow/tensorflow \ && go mod init github.com/tensorflow/tensorflow \
&& sed -i '72 i \ ${TF_DIR}\/tensorflow\/tsl\/protobuf\/*.proto \\' tensorflow/go/genop/generate.sh \ && sed -i '4 i option go_package = "github.com\/tensorflow\/tensorflow\/tensorflow\/go\/core\/framework\/dataset_go_proto";' tensorflow/core/framework/dataset.proto \
&& sed -i '4 i option go_package = "github.com\/tensorflow\/tensorflow\/tensorflow\/go\/core\/framework\/optimized_function_graph_go_proto";' tensorflow/core/framework/optimized_function_graph.proto \
&& sed -i '4 i option go_package = "github.com\/google\/tsl\/tsl\/go\/core\/protobuf\/for_core_protos_go_proto";;' tensorflow/tsl/protobuf/test_log.proto \
&& sed -i '71d;72d' tensorflow/go/genop/generate.sh \
&& sed -i '71 i \ ${TF_DIR}\/tensorflow\/tsl\/protobuf\/*.proto \\' tensorflow/go/genop/generate.sh \
&& sed -i '72 i \ ${TF_DIR}\/tensorflow\/compiler\/xla\/stream_executor\/dnn.proto; do' tensorflow/go/genop/generate.sh \
&& (cd tensorflow/go/op && go generate) \ && (cd tensorflow/go/op && go generate) \
&& go mod edit -require github.com/google/tsl@v0.0.0+incompatible \ && go mod edit -require github.com/google/tsl@v0.0.0+incompatible \
&& go mod edit -replace github.com/google/tsl=/go/src/github.com/google/tsl \ && go mod edit -replace github.com/google/tsl=/go/src/github.com/google/tsl \
...@@ -43,7 +48,7 @@ WORKDIR /example-program ...@@ -43,7 +48,7 @@ WORKDIR /example-program
COPY hello_tf.go . COPY hello_tf.go .
RUN go mod init app \ RUN go mod init app \
&& go mod edit -require github.com/google/tsl@v0.0.0+incompatible \ && go mod edit -require github.com/google/tsl@v0.0.0+incompatible \
&& go mod edit -require github.com/tensorflow/tensorflow@v2.11.0+incompatible \ && go mod edit -require github.com/tensorflow/tensorflow@v2.12.0+incompatible \
&& go mod edit -replace github.com/google/tsl=/go/src/github.com/google/tsl \ && go mod edit -replace github.com/google/tsl=/go/src/github.com/google/tsl \
&& go mod edit -replace github.com/tensorflow/tensorflow=/go/src/github.com/tensorflow/tensorflow \ && go mod edit -replace github.com/tensorflow/tensorflow=/go/src/github.com/tensorflow/tensorflow \
&& go mod tidy \ && go mod tidy \
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment

Impressum Datenschutzerklärung