demo / test.py
lixiang6's picture
Upload 19 files
15f87d2 verified
raw
history blame contribute delete
175 Bytes
import torch
print("CUDA available:", torch.cuda.is_available())
print("cuDNN version:", torch.backends.cudnn.version())
print("cuDNN enabled:", torch.backends.cudnn.enabled)