Browse Source

[docker] add tensornvme in docker (#1354)

* add tensornvme in docker

* fix dockerfile

* fix dockerfile
pull/1355/head
ver217 2 years ago committed by GitHub
parent
commit
38fd8844c0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 7
      docker/Dockerfile

7
docker/Dockerfile

@ -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 .
Loading…
Cancel
Save