Tensorflow2.2安装的坑&GPU支持

2020-10-04  本文已影响0人  梦游的猫头鹰

今天在家里跑tf,发现居然tf.test.is_gpu_available()=False,不能忍,就捣鼓了一下,深坑。

原环境配置:

tensorflow2.0

win10、cuda10.2、cudnn忘了。

原运行结果:tf正常可用,GPU不可用,2060的显卡。

捣鼓半天,配置如下,大家少走点弯路,建议conda搭个python测试环境安装。

tensorflow2.2、keras2.3.1(2.4后不支持多后端了)

win10、python3.7、cuda10.1、cudnn7.6.5、Microsoft Visual C++ 2015-2019 Redistributable(x64)

附:

1、cuda安装流程:https://blog.csdn.net/qq_43427482/article/details/106280151

2、tensorflow与cuda的版本对应:https://www.tensorflow.org/install/source_windows

3、keras与tensorflow的版本对应:https://docs.floydhub.com/guides/environments/

4、官方安装文档,cudnn与网络略有不同,试了下2.2无需理会:https://tensorflow.google.cn/install/gpu

上一篇 下一篇

猜你喜欢

热点阅读