Add poetry support

Author: zkkv

.gitignore

@@ -9,4 +9,9 @@ store/plots
 store/datasets
 store/results
 
-venv
+venv
+.venv
+
+# Normally, you would add this file to VCS but it's ignored here
+# because of different torch GPU accelerators 
+poetry.lock

README.md

@@ -28,17 +28,35 @@ with the following versions of PyTorch and Torchvision:
 * `pytorch 1.11.0`
 * `torchvision 0.12.0`
 
-Further Python-packages used are listed in `requirements.txt`.
-Assuming Python and pip are set up, these packages can be installed using:
-```bash
-pip install -r requirements.txt
-```
+Further Python-packages used are listed in `requirements.txt` and `pyproject.toml`.
 
 The code in this repository itself does not need to be installed, but a number of scripts should be made executable:
 ```bash
 chmod +x main*.py compare*.py all_results.sh
 ```
 
+### Installation via poetry
+If you have poetry installed, you can install all project dependencies with a single command.
+
+For CPU-only PyTorch acceleration:
+```bash
+poetry install --extras "cpu"
+```
+For CUDA PyTorch acceleration:
+```bash
+poetry install --extras "cuda"
+```
+
+The choice of between `cpu` and `cuda` will only affect the source used to install `torch` and `torchvision`.
+
+**Note** that poetry installs dependencies to a virtual environment, not globally.
+
+### Installation via pip
+
+Assuming Python and pip are set up, these packages can be installed using:
+```bash
+pip install -r requirements.txt
+```
 
 ## NeurIPS tutorial "Lifelong Learning Machines"
 This code repository is used for the

pyproject.toml

@@ -0,0 +1,51 @@
+[project]
+name = "continual-learning"
+version = "1.0.0"
+description = "PyTorch implementation of various methods for continual learning (XdG, EWC, SI, LwF, FROMP, DGR, BI-R, ER, A-GEM, iCaRL, Generative Classifier) in three different scenarios."
+authors = [
+    {name = "Gido van de Ven",email = "[email protected]"}
+]
+readme = "README.md"
+requires-python = ">=3.10"
+dependencies = [
+    "numpy (>=2.2.5,<3.0.0)",
+    "scipy (>=1.15.2,<2.0.0)",
+    "pandas (>=2.2.3,<3.0.0)",
+    "tqdm (>=4.67.1,<5.0.0)",
+    "scikit-learn (>=1.6.1,<2.0.0)",
+    "matplotlib (>=3.10.1,<4.0.0)",
+    "visdom (>=0.2.4,<0.3.0)",
+    "jupyterlab (>=4.4.0,<5.0.0)",
+    "ipywidgets (>=8.1.6,<9.0.0)"
+]
+
+[tool.poetry.dependencies]
+torch = [
+    { markers = "extra == 'cpu' and extra != 'cuda'", source = "pytorch-cpu"},
+    { markers = "extra == 'cuda' and extra != 'cpu'", source = "pytorch-cuda"},
+ ]
+torchvision = [
+    { markers = "extra == 'cpu' and extra != 'cuda'", source = "pytorch-cpu"},
+    { markers = "extra == 'cuda' and extra != 'cpu'", source = "pytorch-cuda"},
+ ]
+
+[project.optional-dependencies]
+cpu = ["torch (>=2.6.0)", "torchvision"]
+cuda = ["torch (>=2.6.0)", "torchvision"]
+
+[build-system]
+requires = ["poetry-core>=2.0.0,<3.0.0"]
+build-backend = "poetry.core.masonry.api"
+
+[tool.poetry]
+package-mode = false
+
+[[tool.poetry.source]]
+name = "pytorch-cpu"
+url = "https://download.pytorch.org/whl/cpu"
+priority = "explicit"
+
+[[tool.poetry.source]]
+name = "pytorch-cuda"
+url = "https://download.pytorch.org/whl/cu124"
+priority = "explicit"