From 38fd8844c0c1c2b4c3f32a705580ca53dbfe5f0b Mon Sep 17 00:00:00 2001
From: ver217 <lhx0217@gmail.com>
Date: Thu, 21 Jul 2022 17:44:00 +0800
Subject: [PATCH] [docker] add tensornvme in docker (#1354)

* add tensornvme in docker

* fix dockerfile

* fix dockerfile
---
 docker/Dockerfile | 7 +++++++
 1 file changed, 7 insertions(+)

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