Home > AI

PyTorch快速检测CUDA和cuDNN是否可用的代码片段

linker   ·   发表于 24天前   ·   AI
#测试Pytorch是否下载成功
import torch
#测试cuda能否使用,能使用则返回True
print(torch.cuda.is_available())

#测试cuDNN是否正常,正常返回True
from torch.backends import cudnn
a = torch.tensor(1.)
print(cudnn.is_acceptable(a.cuda()))
0 Reply   |  Until 24天前 | 13 View
LoginCan Publish Content
程序耗时 0.0206秒 内存:363KB