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.

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 as you wish. Here you can modify TensorFlow compilation options.

cd tensorflow/ubuntu-16.04/
# or
# cd tensorflow/centos-6.6

# Build the Docker image
docker-compose build

# Set env variables
export PYTHON_VERSION=3.6
export TF_VERSION_GIT_TAG=v1.9.0
export USE_GPU=0

# 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 as you wish. Here you can modify TensorFlow compilation options.

cd tensorflow/ubuntu-16.04/
# or
# cd tensorflow/centos-6.6

# Build the Docker image
docker-compose build

# Set env variables
export PYTHON_VERSION=3.6
export TF_VERSION_GIT_TAG=v1.9.0
export USE_GPU=1
export CUDA_VERSION=9.1
export CUDNN_VERSION=7.1

# Start the compilation
docker-compose run tf

# You can also do:
# docker-compose run tf bash
# bash build.sh

---

• 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 1.8.0

Py	Distribution	glibc	Processor	Arch	Flags	CUDA	cuDNN	Link
3.6	Ubuntu 16.04	2.23	Intel i7-7700HQ	CPU	avx2 sse	-	-	Link
3.6	Ubuntu 16.04	2.23	Intel i7-7700HQ	GPU	avx2 sse	9.0	7	Link
3.6	Ubuntu 16.04	2.23	Intel i7-7700HQ	GPU	avx2 sse	9.0	7.1	Link
3.6	Ubuntu 16.04	2.23	Intel i7-7700HQ	GPU	avx2 sse	9.1	7.1	Link
3.6	Ubuntu 16.04	2.23	Intel i7-7700HQ	GPU	avx2 sse	9.2	7.1	Link
3.6	Ubuntu 16.04	2.23	Intel i7 960	CPU	avx sse	-	-	Link
3.6	Ubuntu 16.04	2.23	Intel i7 960	GPU	avx sse	9.0	7	Link
3.6	Ubuntu 16.04	2.23	Intel i7 960	GPU	avx sse	9.0	7.1	-
3.6	Ubuntu 16.04	2.23	Intel i7 960	GPU	avx sse	9.1	7.1	Link
3.6	Ubuntu 16.04	2.23	Intel Core i7 960	GPU	avx sse	9.2	7.1	-
3.6	CentOS 6.6	2.12	Intel i7-7700HQ	CPU	avx2 sse	-	-	Link
3.6	CentOS 6.6	2.12	Intel i7-7700HQ	GPU	avx2 sse	9.0	7	-
3.6	CentOS 6.6	2.12	Intel i7-7700HQ	GPU	avx2 sse	9.0	7.1	-
3.6	CentOS 6.6	2.12	Intel i7-7700HQ	GPU	avx2 sse	9.1	7.1	Link
3.6	CentOS 6.6	2.12	Intel i7-7700HQ	GPU	avx2 sse	9.2	7.1	-
3.6	CentOS 6.6	2.12	Intel i7 960	CPU	avx sse	-	-	Link
3.6	CentOS 6.6	2.12	Intel i7 960	GPU	avx sse	9.0	7	-
3.6	CentOS 6.6	2.12	Intel i7 960	GPU	avx sse	9.0	7.1	-
3.6	CentOS 6.6	2.12	Intel i7 960	GPU	avx sse	9.1	7.1	Link
3.6	CentOS 6.6	2.12	Intel i7 960	GPU	avx sse	9.2	7.1	-

Tensorflow 1.9.0

Py	Distribution	glibc	Processor	Arch	Flags	CUDA	cuDNN	Link
3.6	Ubuntu 16.04	2.23	Intel i7 960	CPU	avx sse	-	-	Link
3.6	Ubuntu 16.04	2.23	Intel i7 960	GPU	avx sse	9.1	7.1	Link
3.6	CentOS 6.6	2.12	Intel i7 960	CPU	avx sse	-	-	Link
3.6	CentOS 6.6	2.12	Intel i7 960	GPU	avx sse	9.1	7.1	Link

Authors

• Hadrien Mary [email protected]

License

MIT License. See LICENSE.