如何解决tensorflow gpu测试通过-但我没有安装cuDNN
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?
解决方法
在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.GRU
(Keras GRU layers) 使用了 cuDNN 的功能。
检查 Tensorflow site 中提供的这些示例以了解 GPU 利用率。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。