Skip to content

Commit

Permalink
Install Tensorboard requirements & update versions (lukalabs#2)
Browse files Browse the repository at this point in the history
* Install Tensorflow as it is required by the TensorBoard & up both versions to 1.5.0
  • Loading branch information
nikita-smetanin authored Feb 5, 2018
1 parent d367d42 commit 92b0957
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 4 deletions.
6 changes: 4 additions & 2 deletions dockerfiles/Dockerfile.cpu
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,9 @@

FROM ubuntu:16.04

ARG TENSORBOARD_VERSION=0.1.8
# TensorBoard-related packages versions
ARG TENSORBOARD_VERSION=1.5.0
ARG TENSORFLOW_VERSION=1.5.0

# Install some dependencies
RUN apt-get update && apt-get install -y \
Expand Down Expand Up @@ -34,7 +36,7 @@ RUN git clone https://github.com/lukalabs/cakechat.git /root/cakechat
RUN pip --no-cache-dir install -r /root/cakechat/requirements.txt --process-dependency-links

# setup Tensorboard for monitoring the training process
RUN pip install tensorflow-tensorboard==$TENSORBOARD_VERSION
RUN pip install tensorflow-tensorboard==$TENSORBOARD_VERSION tensorflow==$TENSORFLOW_VERSION
RUN mkdir -p /root/cakechat/data/tensorboard

WORKDIR /root/cakechat
Expand Down
8 changes: 6 additions & 2 deletions dockerfiles/Dockerfile.gpu
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,13 @@

FROM nvidia/cuda:8.0-cudnn5-devel-ubuntu16.04

# Theano-related packages versions
ARG THEANO_VERSION=rel-0.9.0
ARG GPUARRAY_VERSION=v0.6.2
ARG TENSORBOARD_VERSION=0.1.8

# TensorBoard-related packages versions
ARG TENSORBOARD_VERSION=1.5.0
ARG TENSORFLOW_VERSION=1.5.0

# Install some dependencies
RUN apt-get update && apt-get install -y \
Expand Down Expand Up @@ -87,7 +91,7 @@ RUN git clone https://github.com/lukalabs/cakechat.git /root/cakechat
RUN pip --no-cache-dir install -r /root/cakechat/requirements.txt --process-dependency-links

# setup Tensorboard for monitoring the training process
RUN pip install tensorflow-tensorboard==$TENSORBOARD_VERSION
RUN pip install tensorflow-tensorboard==$TENSORBOARD_VERSION tensorflow==$TENSORFLOW_VERSION
RUN mkdir -p /root/cakechat/data/tensorboard

WORKDIR /root/cakechat
Expand Down

0 comments on commit 92b0957

Please sign in to comment.