From 42dd3b7c575e0b245ac171fbb5245db67fdebfe5 Mon Sep 17 00:00:00 2001
From: Bubbles The Dev <152947339+KernFerm@users.noreply.github.com>
Date: Wed, 16 Oct 2024 22:55:58 -0400
Subject: [PATCH] Add files via upload

Signed-off-by: Bubbles The Dev <152947339+KernFerm@users.noreply.github.com>
---
 get_device.py | 6 ++++++
 1 file changed, 6 insertions(+)
 create mode 100644 get_device.py

diff --git a/get_device.py b/get_device.py
new file mode 100644
index 0000000..13e658c
--- /dev/null
+++ b/get_device.py
@@ -0,0 +1,6 @@
+import torch
+
+print(torch.cuda.is_available())  # This will return True if CUDA is available
+print(torch.version.cuda)  # This will print the CUDA version being used
+print(torch.cuda.get_device_name(0))  # This will print the name of the GPU, e.g., 'NVIDIA GeForce RTX GPU Model'
+