[docker] add tensornvme in docker (#1354)

* add tensornvme in docker * fix dockerfile * fix dockerfile
2 years ago · 38fd8844c0
1 changed files with 7 additions and 0 deletions
--- 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 .