github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-03 01:19:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

[docker] add tensornvme in docker (#1354)

Browse Source 
* add tensornvme in docker

* fix dockerfile

* fix dockerfile
This commit is contained in:
ver217
2022-07-21 17:44:00 +08:00
committed by GitHub
parent 6b43c789fd
commit 38fd8844c0
1 changed files with 7 additions and 0 deletions
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
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 .