【工具】验证是否安装cuda和cuDNN

验证是否安装cuda

【验证cuda是否安装方法1】

import torch
print(torch.cuda.is_available())  #返回True则说明已经安装了cuda
#输出
True

【验证cuda是否安装方法2】

win+r–>输入cmd–>回车
在命令行下输入nvcc -V后回车，如果出现如下所示内容说明cuda已经成功安装
如果显示nvcc不是内部或外部命令，也不是可运行的程序或批处理文件。说明cuda没有成功安装

验证cuDNN是否成功安装

其实这里说cuDNN安装是不恰当的，cuDNN不需要安装，只需要下载并把文件放在合适的位置即可，想了解如何安装cuDNN的话可以看我的另外一篇博客：cuDNN安装
【验证cuDNN是否安装】

from torch.backends import cudnn 
print(cudnn.is_available())  #返回True说明已经安装了cuDNN
#输出
True