FROM pytorch/pytorch:latest
ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
RUN apt-get update && apt install libsndfile1 -y
RUN pip install --upgrade pip
RUN pip install future
RUN pip install tensorflow==2.3.0 numpy==1.18.4 inflect==4.1.0 librosa==0.8.0 scipy==1.4.1 Unidecode==1.1.1 pkuseg==0.0.25 tqdm==4.48.0 tensorboardx==2.1 pyyaml==5.3.1 jieba==0.42.1 g2pc==0.9.9.3 pypinyin==0.38.1 matplotlib==3.3.0 kashgari==2.0.0 tensorflow-gpu==2.3.0 pydub==0.24.1 pillow numba==0.48.0 jupyter -i https://pypi.douban.com/simple/
#EXPOSE 8888
#ADD apex /apex/
#WORKDIR /apex/
#RUN python setup.py install
RUN mkdir /app
WORKDIR /app
