tensorflow gpu测试,但我没有安装cuDNN

2024-05-29 02:59:40 发布

您现在位置:Python中文网/ 问答频道 /正文

Windows10 pro,单个RTX 2080 Ti。我是Tensorflow的新手

我刚刚安装了tensorflow gpu,版本2.1.0,python 3.7.7。Cuda编译工具,10.1版,V10.1.105。没有自编的。我还没有安装cuDNN,也没有注册。所有安装都是标准的,没有自我编译

tensorflow.org文档说明使用GPU需要cuDNN。但我的GPU使用测试似乎通过了。例如 tf.config.experimental.list_physical_devices('GPU')返回[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]

看来我应该只安装cuDNN,而不要再失眠了。但是我仍然想知道我是否在使用GPU,所以我更喜欢一个可能失败的测试

是否有真正的测试来确定安装是否会使用GPU


Tags: 工具版本gpudevicetensorflowticudapro
1条回答
网友
1楼 · 发布于 2024-05-29 02:59:40

在NVIDIA GPU计算工具包中,可以验证cuDNN安装, 在windows系统上, 去

C:\Programming Files\NVIDIA GPU computing Toolkit/CUDA\v10.0\include\

open cudnn.h 

要成功利用Tensorflow GPU,需要CUDA和cuDNN。 一些Tensorflow库,如tf.keras.layers.GRUKeras GRU layers)使用了cuDNN的功能

检查Tensorflow site中提供的这些示例,了解GPU的使用情况

相关问题 更多 >

    热门问题