验证环境B
验证成功,没有问题
验证环境A
得到结果false
检查是否与CUDA关联成功
tf.test.is_built_with_cuda()
发现没有关联成功
根据查询可知道,失败的原因是1.13.x版本的tensorflow和目前安装的cuda版本不合适,所以刚才tensorflow2.6版本的成功,而这个1.13.x失败了
对应版本查看:
{ "src": "https://img2022.cnblogs.com/blog/2571021/202202/2571021-20220217165611454-1497389376.png", "status": "error", "percent": 0, "align": "left", "linkTarget": "_blank", "display": "inline", "message": "网络异常,图片无法展示" }
为了让tensorflow 1.13.x 版本也能使用,我们需要安装对应版本的cudn和cudnn
安装教程同【人工智能】【深度学习】CUDA与CUDNN安装 - 萌狼蓝天 - 博客园 (cnblogs.com/mllt)
安装的过程中选择合适版本即可
将下载的内容解压到CUDN安装位置
我的是 C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.0
再进行测试
成功。