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


本文介绍两种方法来验证CUDA是否已成功安装:通过Python的PyTorch库检查和命令行工具nvcc。此外,还提供了一种验证cuDNN是否正确配置的方法,即利用PyTorch库进行检查。
1268

被折叠的 条评论
为什么被折叠?



