4 lines
72 B
Python
4 lines
72 B
Python
|
import torch
|
||
|
|
print(torch.version.cuda)
|
||
|
|
print(torch.cuda.is_available())
|