https://github.com/wamuir/golang-tf

Working golang + tensorflow
https://github.com/wamuir/golang-tf

docker go golang tensorflow tensorflow2

Last synced: 3 months ago
JSON representation

Working golang + tensorflow

• Host: GitHub
• URL: https://github.com/wamuir/golang-tf
• Owner: wamuir
• License: mit
• Created: 2020-11-12T03:24:42.000Z (about 4 years ago)
• Default Branch: main
• Last Pushed: 2022-08-14T11:15:12.000Z (over 2 years ago)
• Last Synced: 2024-10-04T13:07:08.961Z (4 months ago)
• Topics: docker, go, golang, tensorflow, tensorflow2
• Language: Dockerfile
• Homepage: https://hub.docker.com/r/wamuir/golang-tf
• Size: 8.03 MB
• Stars: 22
• Watchers: 3
• Forks: 10
• Open Issues: 2
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# golang-tf

## About

This repository contains Dockerfiles for using Tensorflow in Go.  You can build on your own or pull an image from [wamuir/golang-tf](https://hub.docker.com/r/wamuir/golang-tf/tags?page=1&ordering=name) on Docker Hub:

```sh

docker pull wamuir/golang-tf

```

## Example Usage

See the [example Dockerfile](https://github.com/wamuir/golang-tf/blob/main/example-usage/Dockerfile).

## GPU Support

Installation of the [NVIDIA Container Toolkit](https://github.com/NVIDIA/nvidia-docker) is necessary for GPU use.  Then, expose the GPU(s) to the docker container using the `--gpus all` flag.  Example:

```sh

docker run -it --rm --gpus all wamuir/golang-tf:gpu

```

## Background

For background on the problem, see Tensorflow issues [39307](https://github.com/tensorflow/tensorflow/issues/39307), [39744](https://github.com/tensorflow/tensorflow/issues/39744), [41808](https://github.com/tensorflow/tensorflow/issues/41808), [43847](https://github.com/tensorflow/tensorflow/issues/43847), and [44655](https://github.com/tensorflow/tensorflow/pull/44655), among others.