验证是否安装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