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())
|