diff --git a/Dockerfile b/Dockerfile
index e9065040b52130675e30bc9e6adc79f20850e30f..41007875bdaa25868722ffda35c25d43ebab8c9d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -5,6 +5,7 @@ FROM nvidia/cuda:${cuda_version}-cudnn${cudnn_version}-devel
 # Install system packages
 RUN apt-get update && apt-get install -y --no-install-recommends \
       bzip2 \
+      unzip \
       g++ \
       git \
       graphviz \