Skip to content

GitLab

Unverified Commit 38fd8844 authored by ver217's avatar ver217 Committed by GitHub
Browse files

[docker] add tensornvme in docker (#1354) 

* add tensornvme in docker

* fix dockerfile

* fix dockerfile
parent 6b43c789
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 0 deletions
docker/Dockerfile
View file @ 38fd8844
...@@ -15,3 +15,10 @@ RUN git clone https://github.com/hpcaitech/ColossalAI.git \ ...@@ -15,3 +15,10 @@ RUN git clone https://github.com/hpcaitech/ColossalAI.git \
# install titans # install titans
RUN pip install --no-cache-dir 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
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment