Skip to content

Commit 0b7fb12

Browse files
committed
Switch to tensorflow backend
1 parent 911902f commit 0b7fb12

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

worker/Dockerfile

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,5 +41,6 @@ RUN lein
4141
RUN apt-get install -y git
4242

4343
RUN pip3 install numpy scipy scikit-learn pillow h5py
44-
RUN pip3 install --upgrade --no-deps git+git://github.com/Theano/Theano.git
44+
RUN export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.12.0rc0-cp34-cp34m-linux_x86_64.whl
45+
RUN pip3 install --upgrade $TF_BINARY_URL
4546
RUN pip3 install keras

0 commit comments

Comments
 (0)