[docker] add dockerfile (#31)

* add dockerfile * update readme
hpcaitech · Jul 21, 2022 · 446385c · 446385c
1 parent 63f7df3
commit 446385c
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -55,6 +55,12 @@ pip install tensornvme
 ```
 All acceptable environment variables are the same as those when installing from source.
 
+## Use docker
+
+```shell
+git clone https://github.com/hpcaitech/TensorNVMe.git && cd TensorNVMe/docker && docker build -t tensornvme .
+```
+
 ## CLI
 
 We provide a CLI to test whether backends work well.

diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -0,0 +1,10 @@
+FROM hpcaitech/pytorch-cuda:1.11.0-11.3.0
+
+# install dependencies
+RUN conda install -y cmake
+
+# install tensornvme
+RUN git clone https://github.com/hpcaitech/TensorNVMe.git && \
+    cd TensorNVMe && \
+    pip install -r requirements.txt && \
+    pip install -v --no-cache-dir .