From 53a97e83df2657ef84d4ffff88b4fca92df33a76 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=96=9B=E4=B8=9E=E5=AE=8F?= Date: Fri, 1 Nov 2019 14:21:00 +0800 Subject: [PATCH] [bug-fix] pillow dependency in Dockerfile the same as [requirements.txt](https://github.com/NVIDIA/tacotron2/blob/master/requirements.txt#L9) ``` Traceback (most recent call last): File "train.py", line 290, in args.warm_start, args.n_gpus, args.rank, args.group_name, hparams) File "train.py", line 248, in train hparams.distributed_run, rank) File "train.py", line 146, in validate logger.log_validation(val_loss, model, y, y_pred, iteration) File "/workspace/tacotron2/logger.py", line 34, in log_validation iteration) File "/opt/conda/lib/python3.6/site-packages/tensorboardX/writer.py", line 584, in add_image image(tag, img_tensor, dataformats=dataformats), global_step, walltime) File "/opt/conda/lib/python3.6/site-packages/tensorboardX/summary.py", line 271, in image image = make_image(tensor, rescale=rescale) File "/opt/conda/lib/python3.6/site-packages/tensorboardX/summary.py", line 307, in make_image from PIL import Image ModuleNotFoundError: No module named 'PIL' ``` --- Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index adf4658..44864b9 100644 --- a/Dockerfile +++ b/Dockerfile @@ -3,7 +3,7 @@ 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 +RUN pip install numpy scipy matplotlib librosa==0.6.0 tensorflow tensorboardX inflect==0.2.5 Unidecode==1.0.22 pillow jupyter ADD apex /apex/ WORKDIR /apex/