torch code to inspect version and device availability

app.py

@@ -1,4 +1,17 @@
 import streamlit as st
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+# x = st.slider('Select a value')
+# st.write(x, 'squared is', x * x)
+
+import torch
+
+print(torch.__version__)
+
+st.write("torch version: " + torch.__version__)
+st.write(" ")
+st.write("cuda available: " + torch.cuda.is_available())
+st.write(" ")
+st.write("No of GPUs: " + torch.cuda.device_count())
+st.write(" ")
+st.write("device name: " + torch.cuda.get_device_name())
+