Dockerfile update
-FROM nvidia/cuda:9.1-cudnn7-runtime-ubuntu16.04
+FROM nvidia/cuda:9.0-cudnn7-runtime-ubuntu16.04
MAINTAINER Davide Caroselli <davide@modernmt.eu>
# Install dependencies
@@ -13,7 +13,7 @@ RUN add-apt-repository -y ppa:openjdk-r/ppa
RUN apt-get -y update
RUN apt-get install -y openjdk-8-jdk
-RUN apt-get install -y python-pip python-requests python-psutil
+RUN apt-get install -y python-dev python-pip python-requests python-psutil
# Set ulimit
RUN echo "* soft nofile 1048576" > /etc/security/limits.conf
@@ -28,21 +28,21 @@ RUN localedef -i en_US -c -f UTF-8 en_US.UTF-8
RUN echo "LC_ALL=en_US.UTF-8" >> /etc/environment
RUN echo "LANG=en_US.UTF-8" >> /etc/environment
-# Install PyTorch
-RUN pip install http://download.pytorch.org/whl/cu91/torch-0.3.1-cp27-cp27mu-linux_x86_64.whl
+# Install Tensorflow
+RUN pip install numpy==1.14.5
+RUN pip install tensorflow-gpu==1.8.0
+RUN pip install tensor2tensor==1.6.3
+RUN pip install --upgrade oauth2client
-# Create a new user "mmt" and login as "mmt"
+# Create a new user "ubuntu" and login as "ubuntu"
RUN apt-get install -y sudo && rm -rf /var/lib/apt/lists/*
-RUN useradd mmt
-RUN passwd -d mmt
-RUN usermod -aG sudo mmt
-WORKDIR /home/mmt
-RUN chown mmt:mmt /home/mmt
-USER mmt
-
-# Install MMT
-RUN wget -qO- https://github.com/ModernMT/MMT/releases/download/v2.5/mmt-2.5-ubuntu.tar.gz | tar xvfz -
-WORKDIR /home/mmt/mmt
-
-# Make MMT example engine
-RUN ./mmt create en it examples/data/train
+RUN useradd ubuntu
+RUN passwd -d ubuntu
+RUN usermod -aG sudo ubuntu
+WORKDIR /home/ubuntu
+RUN chown ubuntu:ubuntu /home/ubuntu
+USER ubuntu
+
+# Download MMT
+RUN wget -qO- https://github.com/ModernMT/MMT/releases/download/v3.0/mmt-3.0-ubuntu.tar.gz | tar xvfz -
+WORKDIR /home/ubuntu/mmt
\ No newline at end of file
