验证环境B

image.png

image.png

image.png

验证成功,没有问题

验证环境A

image.png

image.png

得到结果false

检查是否与CUDA关联成功

tf.test.is_built_with_cuda()

image.png

发现没有关联成功

根据查询可知道,失败的原因是1.13.x版本的tensorflow和目前安装的cuda版本不合适,所以刚才tensorflow2.6版本的成功,而这个1.13.x失败了

对应版本查看:

Build from source on Windows | TensorFlow (google.cn)

image.png

image.png

为了让tensorflow 1.13.x 版本也能使用,我们需要安装对应版本的cudn和cudnn

安装教程同【人工智能】【深度学习】CUDA与CUDNN安装 - 萌狼蓝天 - 博客园 (cnblogs.com/mllt)

安装的过程中选择合适版本即可

image.png

image.png

将下载的内容解压到CUDN安装位置

我的是 C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.0

image.png

再进行测试

image.png

成功。

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐