diff --git a/docker/requirements.txt b/docker/requirements.txt
index 33ff92b..977c777 100644
--- a/docker/requirements.txt
+++ b/docker/requirements.txt
@@ -9,4 +9,5 @@ cmapy
 pytorch3d
 tables
 numpy-stl
-torchmetrics
\ No newline at end of file
+torchmetrics
+pytorch_msssim
\ No newline at end of file