Browse Source

Dockerfile: updating to use latest pytorch and apex

master
rafaelvalle 5 years ago
parent
commit
d6670c8ed7
1 changed files with 9 additions and 1 deletions
  1. +9
    -1
      Dockerfile

+ 9
- 1
Dockerfile View File

@ -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" .

Loading…
Cancel
Save