From 6be866b0b8b5ec676159eaa56c09371c02cbd7e1 Mon Sep 17 00:00:00 2001
From: Anthony Wang
Date: Tue, 14 Dec 2021 10:11:33 -0600
Subject: Add PyTorch GPU test script

---
 gpu.py | 5 +++++
 1 file changed, 5 insertions(+)
 create mode 100644 gpu.py

diff --git a/gpu.py b/gpu.py
new file mode 100644
index 0000000..e40cb5f
--- /dev/null
+++ b/gpu.py
@@ -0,0 +1,5 @@
+import torch
+
+print(torch.cuda.is_available())
+print(torch.version.hip)
+cuda = torch.device('cuda')
\ No newline at end of file
-- 
cgit v1.2.3-70-g09d2