add python gpu test

gpu_test.py

@@ -0,0 +1,7 @@
+import torch
+
+print(f"is availeble = {torch.cuda.is_available()}")
+
+print(f"device count = {torch.cuda.device_count()}")
+
+print(f"current device = {torch.cuda.current_device()}")

startup.sh

@@ -8,6 +8,7 @@ echo "LS: $(ls -als)"
 
 nvidia-smi
 
+python gpu_test.py
 python -m serve.controller --host 0.0.0.0 --port 10000 &
 P1=$!