Add PyTorch GPU test script

1 files changed, 5 insertions, 0 deletions

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