Pytorch检查CUDA和cudnn是否可用及其版本

Pytorch检查CUDA和cudnn版本

    • 检查CUDA
    • 检查cudnn

命令行终端启动python

检查CUDA

>>> import torch
>>> torch.cuda.is_available()   # 检查cuda是否可用
>>> torch.version.cuda          # 查看cuda版本

检查cudnn

>>> torch.backends.cudnn.is_available()  # 检查cudnn是否可用
>>> torch.backends.cudnn.version()       # 查看cudnn版本

你可能感兴趣的:(深度学习,pytorch,深度学习,python)