diff --git a/requirements.txt b/requirements.txt
index d323e7a1409..78d0af566f7 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -25,7 +25,7 @@ tensorboard
 jinja2==3.0.3
 pytorch-lightning
 torchx
-torchrl
+torchrl==0.2.0
 ax-platform
 nbformat>=4.2.0
 datasets