实际开发中为了使程序高效,我们常常用 Python训练各种模型然后在 Go 中运行。
TensorFlow 目前提供了一个 Go API,该 API 特别适合加载用 Python 创建的模型并在 Go 应用中运行这些模型。但是TensorFlow Go环境配置起来要比Python复杂很多。
第一步需要配置 TensorFlow C 库(只提供linux和mac版本)
下载库,这里我配置的是1.15版本。
TF_TYPE="cpu" # Change to "gpu" for GPU support
OS="darwin" # Change to "darwin" for Mac OS
TARGET_DIRECTORY="/usr/local" # Change to other folder what you want
curl -L "https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-${TF_TYPE}-${OS}-x86_64-1.15.0.tar.gz" |
sudo tar -C $TARGET_DIRECTORY -xz
设置环境变量
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/lib
export DYLD_LIBRARY_PATH=$DYLD_LIBRARY_PATH:/usr/local/lib
第二步go TensorFlow 安装
go get github.com/tensorflow/tensorflow/tensorflow/go
这一步可能特别慢,甚至超时失败,可以多试几次,或者采用如下方式:
通过 github 下载TensorFlow:(我下载的1.15.0 版本)
https://github.com/tensorflow/tensorflow/releases/tag/v1.15.0
这里需要注意的是下载的TensorFlow 版本一定要和 第一步下载的TensorFlow C库对应。
为确保go generate 顺利执行,先把依赖的包安装好:
go get -v -u github.com/golang/protobuf/proto
go get -v -u github.com/golang/protobuf/protoc-gen-go
然后将下载文件解压后放到 $GOPATH/src/http://github.com/ tensorflow/目录下
回到$GOPATH/src 目录下,执行命令:
go generate github.com/tensorflow/tensorflow/tensorflow/go/op
安装成功后通过如下命令可测试:
go test github.com/tensorflow/tensorflow/tensorflow/go
运行成功输出如下:
ok github.com/tensorflow/tensorflow/tensorflow/go 0.914s
【来思Go】,Let's Go!欢迎关注留言交流学习!