load two layer of cnn

some-test/pytorch_gpu_test.py

@@ -0,0 +1,10 @@
+import torch
+torch.cuda.is_available()
+print(torch.cuda.is_available())
+print(torch.__version__)
+
+
+print(torch.cuda.is_available() )# cuda是否可用
+print(torch.cuda.device_count() )# gpu数量
+print(torch.cuda.current_device())# 当前设备索引, 从0开始
+print(torch.cuda.get_device_name(0))# 返回gpu名字