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