diff --git a/docker/Dockerfile.base b/docker/Dockerfile.base
index 1e235d1..be59043 100644
--- a/docker/Dockerfile.base
+++ b/docker/Dockerfile.base
@@ -128,6 +128,8 @@ RUN cd ~/build && \
 
 RUN cd ~ && rm -rf build
 
+RUN apt-get update && apt-get install -y libyaml-cpp-dev
+
 RUN cd ~ && mkdir Development && cd Development && \
 git clone https://github.com/ceccocats/tkDNN.git && cd tkDNN && \
 mkdir build && cd build && \
@@ -137,4 +139,4 @@ make -j6
 RUN apt-get clean && rm -rf /var/lib/apt/lists/*
 COPY assets/entrypoint_setup.sh /
 ENTRYPOINT ["/entrypoint_setup.sh"]
-CMD ["terminator"]
\ No newline at end of file
+CMD ["terminator"]