diff --git a/openmapflow/Dockerfile b/openmapflow/Dockerfile
index 792deae4..335d3634 100644
--- a/openmapflow/Dockerfile
+++ b/openmapflow/Dockerfile
@@ -1,5 +1,5 @@
 # syntax = docker/dockerfile:experimental
-FROM pytorch/torchserve:0.5.3-cpu
+FROM pytorch/torchserve:0.6.0-cpu
 USER root
 RUN pip3 install --upgrade pip
 ARG VERSION
diff --git a/openmapflow/constants.py b/openmapflow/constants.py
index 6b371398..ee4d0ca4 100644
--- a/openmapflow/constants.py
+++ b/openmapflow/constants.py
@@ -13,7 +13,7 @@
 TEMPLATE_README = TEMPLATES_DIR / "README.md"
 TEMPLATE_DEPLOY_YML = TEMPLATES_DIR / "github-deploy.yaml"
 TEMPLATE_TEST_YML = TEMPLATES_DIR / "github-test.yaml"
-VERSION = "0.2.1rc2"
+VERSION = "0.2.1"
 
 # -------------- Dataframe column names --------------------------------------
 SOURCE = "source"