Debian 9里只有CUDA 8,无法使用最新的tensorflow-gpu和keras。经反复折腾和测试,以下版本可供使用。
- cuDNN 8.0 (需自行在NVidia网站下载)
- NumPy 1.10.4
- Tensorflow-gpu 1.4.1
- Keras 2.1.5
运行Deep Learning With Python书中第一个小程序,屏幕输出如下,
Debian 9里只有CUDA 8,无法使用最新的tensorflow-gpu和keras。经反复折腾和测试,以下版本可供使用。
运行Deep Learning With Python书中第一个小程序,屏幕输出如下,