diff --git a/pyproject.toml b/pyproject.toml
index a598696..691e41a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -38,3 +38,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry.scripts]
 infero = "infero.main:app"
+
+[tool.poetry.extras]
+gpu = ["onnxruntime-gpu"]
+