mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-09 10:02:22 +00:00
12 lines
442 B
Python
Executable File
12 lines
442 B
Python
Executable File
import torch
|
|
import torchaudio
|
|
|
|
print("PyTorch version:", torch.__version__)
|
|
# print("Torchvision version:", torchvision.__version__)
|
|
print("Torchaudio version:", torchaudio.__version__)
|
|
print("CUDA available:", torch.cuda.is_available())
|
|
if torch.cuda.is_available():
|
|
print("CUDA version:", torch.version.cuda)
|
|
print("cuDNN version:", torch.backends.cudnn.version())
|
|
print("GPU name:", torch.cuda.get_device_name(0))
|