hadim / Docker Tensorflow Builder
Licence: mit
Docker images to compile TensorFlow yourself.
Stars: ✭ 155
Programming Languages
shell
77523 projects
Projects that are alternatives of or similar to Docker Tensorflow Builder
Awesome Container Security
Awesome list of resources related to container security
Stars: ✭ 140 (-9.68%)
Mutual labels: docker-image
Docker Hugo
Truly minimal Docker images for Hugo open-source static site generator.
Stars: ✭ 143 (-7.74%)
Mutual labels: docker-image
Portauthority
API that leverages Clair to scan Docker Registries and Kubernetes Clusters for vulnerabilities
Stars: ✭ 148 (-4.52%)
Mutual labels: docker-image
Rust Musl Cross
Docker images for compiling static Rust binaries using musl-cross
Stars: ✭ 143 (-7.74%)
Mutual labels: docker-image
Docker Unms
This image is no longer maintained: https://github.com/oznu/docker-unms/issues/53
Stars: ✭ 145 (-6.45%)
Mutual labels: docker-image
Verdaccio
📦🔐 A lightweight Node.js private proxy registry
Stars: ✭ 12,667 (+8072.26%)
Mutual labels: docker-image
Microsoftmlserver Docker
An unofficial Microsoft Machine Learning Server Docker image.
Stars: ✭ 139 (-10.32%)
Mutual labels: docker-image
Data Science Stack Cookiecutter
🐳📊🤓Cookiecutter template to launch an awesome dockerized Data Science toolstack (incl. Jupyster, Superset, Postgres, Minio, AirFlow & API Star)
Stars: ✭ 153 (-1.29%)
Mutual labels: docker-image
Kafka Manager Docker
CMAK (previous known as Kafka Manager) As Docker Image
Stars: ✭ 145 (-6.45%)
Mutual labels: docker-image
Docker Github Pages
🐳 Alpine Docker Image for GitHub Pages and Jekyll powered sites
Stars: ✭ 147 (-5.16%)
Mutual labels: docker-image
Docker X11 Bridge
Simple Xpra X11 bridge to enable GUI with any docker image
Stars: ✭ 143 (-7.74%)
Mutual labels: docker-image
Wordmove
Multi-stage command line deploy/mirroring and task runner for Wordpress
Stars: ✭ 1,791 (+1055.48%)
Mutual labels: docker-image
Ngx Fastdfs
nginx + lua +fastdfs Real-time dynamic compression of distributed pictures
Stars: ✭ 146 (-5.81%)
Mutual labels: docker-image
Dockerfiles
Phalcon Dockerfiles used for internal purposes.
Stars: ✭ 145 (-6.45%)
Mutual labels: docker-image
Docker Elasticsearch Alpine
Alpine Linux based Elasticsearch Docker Image
Stars: ✭ 154 (-0.65%)
Mutual labels: docker-image
Modern Deep Learning Docker
Modern Deep Learning Docker Image
Stars: ✭ 153 (-1.29%)
Mutual labels: docker-image
Docker Erlang Otp
the Official Erlang OTP image on Docker Hub
Stars: ✭ 147 (-5.16%)
Mutual labels: docker-image
Compile Tensorflow on Docker
Docker images to compile TensorFlow yourself.
Tensorflow only provide a limited set of build and it can be challenging to compile yourself on certain configuration. With this Dockerfile
, you should be able to compile TensorFlow on any Linux platform that run Docker.
Compilation images are provided for Ubuntu 18.10, Ubuntu 16.04, CentOS 7.4 and CentOS 6.6.
Requirements
docker
docker-compose
Usage
- Clone this repository:
git clone https://github.com/hadim/docker-tensorflow-builder.git
TensoFlow CPU
- Edit the
build.sh
file to modify TensorFlow compilation parameters. Then launch the build:
LINUX_DISTRO="ubuntu-16.04"
# or LINUX_DISTRO="ubuntu-18.10"
# or LINUX_DISTRO="centos-7.4"
# or LINUX_DISTRO="centos-6.6"
cd "tensorflow/$LINUX_DISTRO"
# Set env variables
export PYTHON_VERSION=3.6
export TF_VERSION_GIT_TAG=v1.13.1
export BAZEL_VERSION=0.19
export USE_GPU=0
# Build the Docker image
docker-compose build
# Start the compilation
docker-compose run tf
# You can also do:
# docker-compose run tf bash
# bash build.sh
TensorFlow GPU
- Edit the
build.sh
file to modify TensorFlow compilation parameters. Then launch the build:
LINUX_DISTRO="ubuntu-16.04"
# or LINUX_DISTRO="ubuntu-18.10"
# or LINUX_DISTRO="centos-7.4"
# or LINUX_DISTRO="centos-6.6"
cd "tensorflow/$LINUX_DISTRO"
# Set env variables
export PYTHON_VERSION=3.6
export TF_VERSION_GIT_TAG=v1.13.1
export BAZEL_VERSION=0.19
export USE_GPU=1
export CUDA_VERSION=10.0
export CUDNN_VERSION=7.5
export NCCL_VERSION=2.4
# Build the Docker image
docker-compose build
# Start the compilation
docker-compose run tf
# You can also do:
# docker-compose run tf bash
# bash build.sh
- Refer to tested build configurations to know which
BAZEL_VERSION
you need. - Be patient, the compilation can be long.
- Enjoy your Python wheels in the
wheels/
folder. - Don't forget to remove the container to free the space after the build:
docker-compose rm --force
.
Builds
Tensorflow | Python | Distribution | Bazel | CUDA | cuDNN | NCCL | Comment |
---|---|---|---|---|---|---|---|
v2.0.0-alpha0 | 3.6 | Ubuntu 18.10 | 0.20 | 10.0 | 7.5 | 2.4 | seg fault error |
v2.0.0-alpha0 | 3.6 | Ubuntu 18.10 | 0.20 | - | - | - | OK |
v2.0.0-alpha0 | 3.6 | Ubuntu 16.04 | 0.20 | 10.0 | 7.5 | 2.4 | TODO |
v2.0.0-alpha0 | 3.6 | Ubuntu 16.04 | 0.20 | - | - | - | TODO |
1.9.0 | 3.6 | Ubuntu 16.04 | - | - | 0.19 | - | OK |
1.9.0 | 3.6 | Ubuntu 16.04 | 9.0 | 0.19 | 7.1 | - | OK |
1.9.0 | 3.6 | Ubuntu 16.04 | 9.1 | 0.19 | 7.1 | - | OK |
1.9.0 | 3.6 | Ubuntu 16.04 | 9.2 | 0.19 | 7.1 | - | OK |
1.9.0 | 3.6 | CentOS 6.6 | - | - | 0.19 | - | OK |
1.9.0 | 3.6 | CentOS 6.6 | 9.0 | 0.19 | 7.1 | - | OK |
1.9.0 | 3.6 | CentOS 6.6 | 9.1 | 0.19 | 7.1 | - | OK |
1.9.0 | 3.6 | CentOS 6.6 | 9.2 | 0.19 | 7.1 | - | OK |
Authors
- Hadrien Mary [email protected]
License
MIT License. See LICENSE.
Note that the project description data, including the texts, logos, images, and/or trademarks,
for each open source project belongs to its rightful owner.
If you wish to add or remove any projects, please contact us at [email protected].