7 lines
121 B
Python
7 lines
121 B
Python
|
import torch
|
||
|
|
|
||
|
|
x = torch.rand(5,3)
|
||
|
|
print("Random Tensor data: ", x)
|
||
|
|
|
||
|
|
print("Cuda available: ", torch.cuda.is_available())
|