5 lines
99 B
Python
5 lines
99 B
Python
|
import torch
|
||
|
|
||
|
print(torch.cuda.is_available())
|
||
|
print(torch.version.hip)
|
||
|
cuda = torch.device('cuda')
|