[docker] add dockerfile (#31)

* add dockerfile

* update readme

Author: ver217

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.

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 .