5 lines
90 B
Python
5 lines
90 B
Python
import torch
|
|
|
|
avail=torch.cuda.is_available()
|
|
print(f'Is torch running on GPU?: {avail}')
|