Reflect TF's own build instructions (from their Dockerfile). · HadokenCode/docker-python@fac0bf0 · GitHub

HadokenCode / docker-python Public

forked from Kaggle/docker-python

• Notifications You must be signed in to change notification settings
• Fork 0
• Star 0

• Code
• Pull requests
• Actions
• Projects
• Wiki
• Security and quality
• Insights

Additional navigation options

Commit fac0bf0

Browse filesBrowse the repository at this point in the historyBrowse files

Seb Boving

committed

Reflect TF's own build instructions (from their Dockerfile).

1 parent da9877f commit fac0bf0
Copy full SHA for fac0bf0

1 file changed

+3-2Lines changed: 3 additions & 2 deletions

File tree

Expand file treeCollapse file tree

Open diff view settings

Filter options

• Dockerfile

Expand file treeCollapse file tree

Open diff view settings

Collapse file

‎Dockerfile‎

Copy file name to clipboardExpand all lines: Dockerfile

+3-2Lines changed: 3 additions & 2 deletions

Original file line number	Diff line number	Diff line change
@@ -87,6 +87,7 @@ ENV TF_NEED_CUDA=1
87	87	ENV TF_CUDA_VERSION=9.1
88	88	# Precompile for Tesla k80 and p100. See https://developer.nvidia.com/cuda-gpus.
89	89	ENV TF_CUDA_COMPUTE_CAPABILITIES=3.7,6.0
	90	+ ENV TF_CUDNN_VERSION=7
90	91	RUN apt-get install -y python-software-properties zip && \
91	92	echo "deb http://ppa.launchpad.net/webupd8team/java/ubuntu precise main" | tee -a /etc/apt/sources.list && \
92	93	echo "deb-src http://ppa.launchpad.net/webupd8team/java/ubuntu precise main" | tee -a /etc/apt/sources.list && \
@@ -99,8 +100,8 @@ RUN apt-get install -y python-software-properties zip && \
99	100	curl https://bazel.build/bazel-release.pub.gpg | apt-key add - && \
100	101	apt-get update && apt-get install -y bazel && apt-get upgrade -y bazel && \
101	102	cd /usr/local/src && git clone https://github.com/tensorflow/tensorflow && \
102		- cd tensorflow && echo "\n" | ./configure && \
103		- bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package && \
	103	+ cd tensorflow && cat /dev/null | ./configure && \
	104	+ bazel build --config=opt --config=cuda --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" //tensorflow/tools/pip_package:build_pip_package && \
104	105	bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg && \
105	106	pip install /tmp/tensorflow_pkg/tensorflow*.whl && \
106	107	rm -Rf /tmp/tensorflow_pkg

0 commit comments

Comments

0 (0)