Skip to content

Commit af6689e

Browse files
author
Alireza Khodabakhsh
committed
torch version added
1 parent 679544d commit af6689e

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

index.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import torch
2+
print(torch.__version__)
23
dtype = torch.float
34
device = torch.device("cuda" if torch.cuda.is_available() else 'cpu')
45
A=torch.tensor([1,2,3], dtype = dtype, device=device)

0 commit comments

Comments
 (0)