diff --git a/setup.py b/setup.py
index 6b189fcc..db10ecab 100644
--- a/setup.py
+++ b/setup.py
@@ -34,7 +34,7 @@ def run(self):
 requirements = [
     'numpy',
     'packaging',
-    'protobuf>=3.8.0,<4',
+    'protobuf>=3.8.0,<6',
 ]
 
 
diff --git a/test-requirements.txt b/test-requirements.txt
index b966321c..fd21df49 100644
--- a/test-requirements.txt
+++ b/test-requirements.txt
@@ -2,7 +2,7 @@ flake8
 pytest
 torch
 torchvision
-protobuf==3.20.3
+protobuf==5.29.6
 numpy==1.22.0
 tensorboard
 boto3