如何检验cudnn安装成功
时间: 2023-04-09 16:03:03 浏览: 2481
cudnn验证程序
您可以通过运行一个简单的程序来检验cudnn是否安装成功。您可以使用以下代码:
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
如果cudnn已经成功安装,则会在控制台输出类似于以下内容的信息:
I tensorflow/core/common_runtime/gpu/gpu_device.cc:1105] Found device 0 with properties:
name: GeForce GTX 1080 Ti major: 6 minor: 1 memoryClockRate(GHz): 1.582
pciBusID: 0000:01:00.0
totalMemory: 10.91GiB freeMemory: 10.75GiB
I tensorflow/core/common_runtime/gpu/gpu_device.cc:1195] Creating TensorFlow device (/device:GPU:0) -> (device: 0, name: GeForce GTX 1080 Ti, pci bus id: 0000:01:00.0, compute capability: 6.1)
如果您没有看到类似的输出,则表示cudnn未成功安装。
阅读全文