6 lines
103 B
Python
6 lines
103 B
Python
import torch
|
|
print(torch.__version__)
|
|
|
|
print(torch.version.cuda)
|
|
print(torch.backends.cudnn.version())
|