依次输入下面的指令：
import tensorflow as tf
tf.test.is_gpu_available()
提示True，则运行通过。
也可以试试新指令
tf.config.list_physical_devices('GPU')
如果列表不为空，则运行通过。
from tensorflow.keras import layers
x = tf.random.normal([4, 32, 32, 3])
layer = layers.Conv2D(16, kernel_size=3)
out = layer(x)
如果没有ERORR报错，则cudnn测试通过。
Ubuntu 16.04下NVIDIA驱动安装、CUDA安装、cuDNN安装文档参考： cuda、cudnn安装（...
硬件支撑 安装显卡驱动 CUDA安装 cuDNN安装 测试CUDA是否成功 1.硬件支撑 win7 64位 显卡N...
测试cuda与cudnn安装 在cmd命令行模式，输入python，进行python命令行模式。 依次输入下面的指...
CUDA cudnn 安装
安装cuda 官网下载cuda，本文安装的是9.0版本。 下载cudnn 下载对应cuda9.0的cudnn，本文...
无GPU 一行命令即可，然后在python脚本下测试 没有报错的话则安装成功 有GPU 安装CUDA与cuDNN ...
查看系统信息 cuda 和 cudnn 安装 在manjaro下安装cuda 和 cudnn 非常简单： 一般来说...
使用支持gpu的tensorflow的前提是安装了正确版本的CUDA和cuDNN。 关于CUDA和cuDNN的安装...
实验室服务器安装了cuda之后，需要安装cudnn加速，安装记录如下： 下载对应cuda版本的cudnn，需要注册...
为什么是非root用户安装cuda和cudnn？ 共用服务器时，系统公共环境中安装的cuda和cudnn的版本往往...
本文标题：测试cuda与cudnn安装
本文链接：https://www.haomeiwen.com/subject/wlbrxktx.html
网友评论