diff --git a/dev-requirements.txt b/dev-requirements.txt
index 9d82cac567..30463a7562 100644
--- a/dev-requirements.txt
+++ b/dev-requirements.txt
@@ -1,5 +1,10 @@
 parameterized
 pytest
+
+--pre
+--index-url https://download.pytorch.org/whl/nightly/cpu
+torch
+
 pytest-cov
 torcheval-nightly
 torchsnapshot-nightly
diff --git a/tests/framework/test_auto_unit.py b/tests/framework/test_auto_unit.py
index 4b43fc07be..7d911a440a 100644
--- a/tests/framework/test_auto_unit.py
+++ b/tests/framework/test_auto_unit.py
@@ -16,7 +16,6 @@
 COMPILE_AVAIL = False
 if is_torch_version_geq_1_13():
     COMPILE_AVAIL = True
-    import torch._dynamo
 
 from torch.distributed import GradBucket, launcher
 from torch.distributed.fsdp import FullyShardedDataParallel as FSDP