diff --git a/Dockerfile b/Dockerfile
index bf140ce..f3b2e85 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -5,7 +5,7 @@ LABEL com.nvidia.volumes.needed="nvidia_driver"
 ENV PATH /usr/local/nvidia/bin:${PATH}
 ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64:${LD_LIBRARY_PATH}
 
-RUN apt-get update&&apt-get install -y git software-properties-common python-software-properties python-setuptools python3-setuptools
+RUN apt-get update&&apt-get install -y git software-properties-common python-software-properties python-setuptools python3-setuptools python-migrate && apt-get clean all
 
 RUN git clone --recursive http://git.tiker.net/trees/pycuda.git
 RUN git clone --recursive https://github.com/twobombs/cuburn.git