zoukankan      html  css  js  c++  java
  • Hello TensorFlow 三 (Golang)

    在一台ubuntu 16.04.2虚拟机上为golang安装TensorFlow。

    官方参考:https://www.tensorflow.org/install/install_go

    首先安装go 1.9 (参考 https://golang.org/doc/install)

    下载1.9安装包

    wget https://storage.googleapis.com/golang/go1.9.linux-amd64.tar.gz     

    解压到/usr/local

    sudo tar -C /usr/local -xzf go1.9.linux-amd64.tar.gz

    添加环境变量,编辑/etc/profile,

    sudo gedit /etc/profile

    在文件最后加入如下内容:

    export PATH=$PATH:/usr/local/go/bin
    export GOPATH=/home/dell/mygo

    更新环境变量:

    source /etc/profile

    或重启虚拟机,否则每次打开终端,都要执行下source /etc/profile

    接下来安装tensorflow C库

    TF_TYPE="cpu" # Change to "gpu" for GPU support
    wget "https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-${TF_TYPE}-$(go env GOOS)-x86_64-1.3.0.tar.gz"
    sudo tar -C /usr/local -xzf "libtensorflow-${TF_TYPE}-$(go env GOOS)-x86_64-1.3.0.tar.gz"

    解压后,/usr/local中出现两个文件夹:
    /usr/local/lib
    /usr/local/include

    配置链接器

    sudo ldconfig 

    最后下载go包

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

    执行go test 验证 

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

    Hello World

    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())
    }

    编译运行:

    输出了一些指令集相关的告警,最后一行是输出结果。

  • 相关阅读:
    Go 函数方法
    rz上传文件乱码
    pip问题:ImportError: cannot import name main
    docker启动报错 docker: Error response from daemon: OCI runtime create failed: container_linux.go:348
    python遍历列表删除多个元素的坑
    python写入csv文件时的乱码问题
    python中的exec()函数和eval()函数
    tornado的IOLoop.instance()方法和IOLoop.current()方法区别
    安装mysql-python时报错
    python将科学计数法表示的数值的字符串转换成数值型数据
  • 原文地址:https://www.cnblogs.com/majianguo/p/7588357.html
Copyright © 2011-2022 走看看