Adding initial Dockerfile (#108)

Dockerfile

@@ -0,0 +1,15 @@
+FROM pytorch/pytorch:1.13.1-cuda11.6-cudnn8-runtime
+
+ENV DEBIAN_FRONTEND=noninteractive
+
+ADD torchsig/ /build/torchsig
+
+ADD pyproject.toml /build/pyproject.toml
+
+RUN pip3 install /build
+
+RUN pip3 install notebook
+
+WORKDIR /workspace/code
+
+ADD examples/ /workspace/code/examples

README.md

@@ -43,6 +43,17 @@ cd torchsig
 pip install .
 ```
 
+## Using the Dockerfile
+If you have Docker installed along with compatible GPUs and drivers, you can try:
+
+```
+docker build -t torchsig -f Dockerfile .
+docker run -d --rm --network=host --shm-size=32g --gpus all --name torchsig_workspace torchsig tail -f /dev/null
+docker exec torchsig_workspace jupyter notebook --allow-root --ip=0.0.0.0 --no-browser
+```
+
+Then use the URL in the output in your browser to run the examples and notebooks.
+
 ## License
 ---
 TorchSig is released under the MIT License. The MIT license is a popular open-source software license enabling free use, redistribution, and modifications, even for commercial purposes, provided the license is included in all copies or substantial portions of the software. TorchSig has no connection to MIT, other than through the use of this license.

pyproject.toml

@@ -12,7 +12,7 @@ readme = "README.md"
 requires-python = ">=3.7"
 license = {text = "MIT"}
 dependencies = [
-    "torch==1.13.0",
+    "torch==1.13.1",
     "torchvision",
     "tqdm",
     "numpy",