diff --git a/source/lmp/tests/test_lammps_dpa_jax.py b/source/lmp/tests/test_lammps_dpa_jax.py
index 10428b2374..cd295e19f9 100644
--- a/source/lmp/tests/test_lammps_dpa_jax.py
+++ b/source/lmp/tests/test_lammps_dpa_jax.py
@@ -19,6 +19,12 @@
     write_lmp_data,
 )
 
+if os.environ.get("CUDA_VISIBLE_DEVICES", "") != "":
+    pytest.skip(
+        reason="The model is generated with CPU",
+        allow_module_level=True,
+    )
+
 pbtxt_file2 = (
     Path(__file__).parent.parent.parent / "tests" / "infer" / "deeppot-1.pbtxt"
 )