wuchendi commited on
Commit
e7a6c95
·
verified ·
1 Parent(s): f59dca1

Create test.py

Browse files
Files changed (1) hide show
  1. test.py +6 -0
test.py ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ import torch
2
+ x = torch.rand(5, 3)
3
+ print(x)
4
+
5
+ print(torch.__version__)
6
+ print(torch.cuda.is_available())