meow

Author: deependujha

tests/tests_pytorch/strategies/test_fsdp2.py

@@ -440,9 +440,9 @@ def configure_optimizers(self):
     trainer.fit(model, ckpt_path=checkpoint_path_full)
 
 
-@RunIf(max_torch="2.5")
+@RunIf(max_torch="2.5", min_cuda_gpus=1)
 @pytest.mark.parametrize("strategy", ["fsdp2", "fsdp2_cpu_offload"])
 def test_fsdp2_requires_torch_2_6_or_newer(tmp_path, strategy):
     """FSDP2 strategies should error on torch < 2.6."""
     with pytest.raises(ValueError, match="FSDP2Strategy requires torch>=2.6.0."):
-        Trainer(accelerator="cpu", default_root_dir=tmp_path, fast_dev_run=True, strategy=strategy)
+        Trainer(default_root_dir=tmp_path, fast_dev_run=True, strategy=strategy)