diff --git a/device_check.py b/device_check.py
new file mode 100644
index 0000000..00f5fa4
--- /dev/null
+++ b/device_check.py
@@ -0,0 +1,8 @@
+import torch
+
+def main():
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    print(f"Device: {device}")
+
+if __name__ == "__main__":
+    main()
\ No newline at end of file