Sharfraz commited on
Commit
f35f8d1
·
verified ·
1 Parent(s): e76ed01

Create test pytorch

Browse files
Files changed (1) hide show
  1. test pytorch +3 -0
test pytorch ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ import torch
2
+ print(torch.__version__)
3
+ print("CUDA available:", torch.cuda.is_available())