diff --git a/requirements.txt b/requirements.txt
index 9bf2ac45..09dc3e5a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,3 @@
 torch
-labml
+labml>=0.4.37
 labml-helpers
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 2bd4755d..ada8f50d 100644
--- a/setup.py
+++ b/setup.py
@@ -17,7 +17,7 @@ setuptools.setup(
     },
     packages=setuptools.find_packages(exclude=('test',
                                                'test.*')),
-    install_requires=['labml',
+    install_requires=['labml>=0.4.37',
                       'labml_helpers',
                       'torch'],
     classifiers=[