diff --git a/Dockerfile b/Dockerfile index f211fa4..adf4658 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,2 +1,10 @@ -FROM pytorch/pytorch:0.4_cuda9_cudnn7 +FROM pytorch/pytorch:nightly-devel-cuda10.0-cudnn7 +ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH} + +RUN apt-get update -y + RUN pip install numpy scipy matplotlib librosa==0.6.0 tensorflow tensorboardX inflect==0.2.5 Unidecode==1.0.22 jupyter + +ADD apex /apex/ +WORKDIR /apex/ +RUN pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" .