TensorFlow、Keras、CUDA、CuDNN 之间的版本要求

文章目录
  1. 1. TensorFlow 、CUDA、CuDNN
  2. 2. TensorFlow、Keras
  3. 3. pip 安装指定版本

配置 Keras 深度学习环境时需要注意 TensorFlow、Keras、CuDNN(GPU) 版本间的版本配对问题。

TensorFlow 、CUDA、CuDNN

Linux 下:

Windows 下:

TensorFlow、Keras

pip 安装指定版本

1
2
pip install tensorflow-gpu==1.2.0
pip install keras==2.1.1 -U --pre