pytorch 命令

1.检查cuda

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

2.检查cudnn

torch.backends.cudnn.is_available()  # 检查cudnn是否可用
torch.backends.cudnn.version()       # 查看cudnn版本
posted @ 2022-07-12 18:57  无左无右  阅读(38)  评论(0编辑  收藏  举报