diff --git a/Dockerfile_cuda b/Dockerfile_cuda
index fdf6be8..47e3a75 100644
--- a/Dockerfile_cuda
+++ b/Dockerfile_cuda
@@ -99,6 +99,9 @@ RUN source ~/.bashrc \
 RUN source ~/.bashrc \
 && pip3 install -q mp_api
 
+RUN source ~/.bashrc \
+&& pip3 install -q astropy
+
 RUN apt-get update \
 && apt-get install -y libgl1-mesa-glx
 
diff --git a/Dockerfile_mpich b/Dockerfile_mpich
index e4cb30a..efa250d 100644
--- a/Dockerfile_mpich
+++ b/Dockerfile_mpich
@@ -94,6 +94,9 @@ RUN source ~/.bashrc \
 RUN source ~/.bashrc \
 && pip3 install -q mp_api
 
+RUN source ~/.bashrc \
+&& pip3 install -q astropy
+
 RUN apt-get update \
 && apt-get install -y libgl1-mesa-glx