diff --git a/Dockerfile b/Dockerfile
index 6944e76..b00b17d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -11,7 +11,7 @@ FROM nvcr.io/nvidia/pytorch:21.08-py3
 ENV PYTHONDONTWRITEBYTECODE 1
 ENV PYTHONUNBUFFERED 1
 
-RUN pip install imageio-ffmpeg==0.4.4 pyspng==0.1.0
+RUN pip install imageio imageio-ffmpeg==0.4.4 pyspng==0.1.0
 
 WORKDIR /workspace