[Перевод] Установите версию Tensorflow для Go

Google искусственный интеллект TensorFlow
[Перевод] Установите версию Tensorflow для Go

Эта статья взята изПрограмма перевода самородковОрганизуется переводОфициальная документация TensorFlow. Если вам интересно, добро пожаловатьПодать заявку на переводчика, изучив учебник переводчика, участвуйте в переводе и сопоставлении статей и документов. Мы также набираем переводчиков TensorFlow, поэтому, пожалуйста, примите активное участие.

TensorFlow предоставляет API, которые можно вызывать из программ Go. Эти API отлично подходят для загрузки моделей, созданных на Python, и их выполнения в приложениях Go. В этой статье будет описано, как установить и настроитьПакет TensorFlow Go.

предупреждать:API языка TensorFlow Go не является частью TensorFlow.Гарантия стабильности API.

Поддерживаемые платформы

Вы можете установить Go-версию TensorFlow в следующих операционных системах:

  • Linux
  • Mac OS X

Установить

Версия TensorFlow для Go основана на библиотеке TensorFlow C. Выполните следующие шаги, чтобы установить эту библиотеку и включить TensorFlow:

  1. Решите, включать ли только ЦП или вместе с графическим процессором при запуске TensorFlow. Чтобы помочь вам принять это решение, прочитайте раздел «Выбор TensorFlow для установки» в следующем руководстве:

    • @{install_linux#determine_which_tensorflow_to_installУстановите TensorFlow в Linux}
    • @{install_mac#determine_which_tensorflow_to_installУстановите TensorFlow в macOS}

  2. Загрузите и разархивируйте библиотеку TensorFlow C, выполнив следующие команды, чтобы/usr/local/libсодержание:

    TF_TYPE="cpu" # Change to "gpu" for GPU support
TARGET_DIRECTORY='/usr/local'
curl -L \
  "https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-${TF_TYPE}-$(go env GOOS)-x86_64-1.4.0-rc0.tar.gz" |
sudo tar -C $TARGET_DIRECTORY -xz

    tarКоманда извлечет библиотеку TensorFlow C вTARGET_DIRECTORYподкаталогlib. Например, укажите/usr/localв видеTARGET_DIRECTORYсделатьtarкоманда для распаковки библиотеки TensorFlow C в/usr/local/lib. Если вы хотите извлечь файлы библиотеки в другой каталог, заменитеTARGET_DIRECTORYВот и все.

  3. На втором этапе, если вы указали системный каталог (например,/usr/local) в видеTARGET_DIRECTORY, то вам нужно запуститьldconfigдля настройки ссылки. Например:

    sudo ldconfig

    если вы укажетеTARGET_DIRECTORYне является системным каталогом (например,~/mydir), тогда вам нужно распаковать каталог (например,~/mydir/lib) к следующим двум переменным среды:

     export LIBRARY_PATH=$LIBRARY_PATH:~/mydir/lib # 用于 Linux 和 Mac OS X
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:~/mydir/lib # 仅用于 Linux
export DYLD_LIBRARY_PATH=$DYLD_LIBRARY_PATH:~/mydir/lib # 仅用于 Mac OS X

  4. Теперь, когда библиотека TensorFlow C установлена, выполнитеgo getЧтобы загрузить соответствующий пакет и соответствующие зависимости:

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

  5. воплощать в жизньgo testЧтобы убедиться, что версия TensorFlow для Go успешно установлена:

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

еслиgo getилиgo testВыдается сообщение об ошибке, вы можетеStackOverflowИщите и задавайте вопросы выше для возможных решений.

Hello World

После установки версии TensorFlow для Go перейдите к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())
}

Дополнительные примеры языка TensorFlow Go см.example in the API documentation, в этом примере используется предварительно обученная модель TensorFlow для маркировки содержимого изображения.

бегать

Запустите, вызвав следующую командуhello_tf.go:

go run hello_tf.go
Hello from TensorFlow version number

Программа может выводить предупреждения, подобные следующим, которые вы можете игнорировать:

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.

Скомпилировать из исходников

TensorFlow имеет открытый исходный код. вы можете следить за этимотдельный документИнструкции в разделе Компиляция версии TensorFlow для Go с использованием исходного кода TensorFlow.

Программа перевода самородковэто сообщество, которое переводит высококачественные технические статьи из Интернета сНаггетсДелитесь статьями на английском языке на . Охват контентаAndroid,iOS,внешний интерфейс,задняя часть,блокчейн,продукт,дизайн,искусственный интеллектЕсли вы хотите видеть более качественные переводы, пожалуйста, продолжайте обращать вниманиеПрограмма перевода самородков,официальный Вейбо,Знай колонку.

Категории