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