安装tensorflow-gpu
参见
https://github.com/tensorflow/tensorflow/issues/26254
TF installation via pip is pointing to 1.13.1, requiring CUDA 10, while in the documentation it’s mentioned CUDA 9
但是我们安装的是CUDA 9.0
In order to install via pip for cuda 9 one should
1 | pip install tensorflow-gpu==1.12.0 |