6 lines
69 B
Python
6 lines
69 B
Python
import torch
|
|
|
|
is_gpu = torch.cuda.is_available()
|
|
|
|
print(str(is_gpu))
|