安装tensorflow_Go 语言TensorFlow 配置安装

8175fad982acbb2fbe62ab61ad43fe7c.png

实际开发中为了使程序高效,我们常常用 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!欢迎关注留言交流学习!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值