本文配置
python 环境:python 3.6
cuda 版本号:cuda_10.0.130_411.31_win10
cudnn 版本号:cudnn-10.0-windows10-x64-v7.6.5.32
Tensorflow 版本号: tensorflow-gpu 1.14.0
Keras 版本号:keras 2.2.5
写在前面
在电脑里找到NIVIDIA Control Pannel查看你的电脑能支持的cuda最高版本,如果安装了高于此版本的cuda,即使安装成功也无法正常使用。
比如,如果按照tensorflw-cpu版本的方法来安装gpu版本,即,在虚拟环境中:conda install tensorflow-gpu
在我这里显示会直接安装tensorflow2.1 + cuda10.1.243+cudnn7.6.5,就会导致明明安装成功但tensorflow使用时出错的问题。
另外,确定了cuda版本之后,后续安装的 cuda、cudnn、Tensorflow、keras 的版本号务必匹配,否则会发生意想不到的错误!!!
TensorFlow和cuda以及cudnn版本匹配查询
https://tensorflow.google.cn/install/source_windows
Tensorflow 与Keras版本匹配查询
https://docs.floydhub.com/guides/environments/