diff --git a/.gitignore b/.gitignore
index 0843976c6a7ed58d7d43fae5ed3746b8eed29783..7368bbbb54b8877fbce97bab543a2f1069fa20e9 100644
--- a/.gitignore
+++ b/.gitignore
@@ -4,3 +4,4 @@ __pycache__
 *.png
 .ipynb_checkpoints
 postcards_256/*
+venv
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000000000000000000000000000000000000..97bf5f5000457b37f9a43069be9d776eb5c2fb4d
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,6 @@
+torch
+scikit-image
+matplotlib
+numpy
+pillow
+requests
diff --git a/train_colorizer.ipynb b/train_colorizer.ipynb
index 95662f1a247e705d18d02d0c93ebee4ef630b80e..d2bc4626acb0f996d9de3f445b69be0804d939ab 100644
--- a/train_colorizer.ipynb
+++ b/train_colorizer.ipynb
@@ -17,7 +17,6 @@
    "source": [
     "import os\n",
     "import requests\n",
-    "import torch\n",
     "from torch.utils.data import random_split\n",
     "from torch.utils.data import DataLoader\n",
     "from models.ColDataset import ColDataset\n",
@@ -171,6 +170,7 @@
    ],
    "source": [
     "# Get a random AKON postcard\n",
+    "import torch\n",
     "from random import randint\n",
     "from skimage import io\n",
     "from iiif_utils import get_manifesturls_from_collectionurl, get_imgurls_from_manifesturl\n",