github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[docker] add tensornvme in docker (#1354) 

* add tensornvme in docker

* fix dockerfile

* fix dockerfile
pull/1355/head
ver217 2022-07-21 17:44:00 +08:00 committed by GitHub
parent 6b43c789fd
commit 38fd8844c0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
docker/Dockerfile
View File

@ -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 .