diff --git a/docker/Dockerfile b/docker/Dockerfile
index cd4b9dc62..4b55dc1eb 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -15,3 +15,10 @@ RUN git clone https://github.com/hpcaitech/ColossalAI.git \
 
 # install titans
 RUN pip install --no-cache-dir titans
+
+# install tensornvme
+RUN conda install cmake && \
+    git clone https://github.com/hpcaitech/TensorNVMe.git && \
+    cd TensorNVMe && \
+    pip install -r requirements.txt && \
+    pip install -v --no-cache-dir .
\ No newline at end of file