update: test_cpu_offload_optimizer

kozistr · Oct 26, 2024 · 26eb948 · 26eb948
1 parent 23e0fda
commit 26eb948
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -246,6 +246,9 @@ def test_version_utils():
 
 
 def test_cpu_offload_optimizer():
+    if not torch.cuda.is_available():
+        pytest.skip('need GPU to run a test')
+
     params = Example().parameters()
 
     opt = CPUOffloadOptimizer(params, torch.optim.AdamW, fused=False, offload_gradients=True)