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系统上, 去
要成功利用Tensorflow GPU,需要CUDA和cuDNN。 一些Tensorflow库,如
tf.keras.layers.GRU
(Keras GRU layers)使用了cuDNN的功能检查Tensorflow site中提供的这些示例,了解GPU的使用情况
相关问题 更多 >
编程相关推荐