1
0
PytorchTutorials/working_install_test.py

7 lines
121 B
Python
Raw Permalink Normal View History

import torch
x = torch.rand(5,3)
print("Random Tensor data: ", x)
print("Cuda available: ", torch.cuda.is_available())