PyCharm 中的 Keras 不使用 GPU

Question

这些线程没有解决我的问题： Keras does not use GPU on Pycharmhaving python 3.5 and Tensorflow 1.4

Keras 与 TensorFlow 后端不使用 GPU

我已经在运行 Windows 10 并具有 GTX 750 Ti 显卡的 PC 上安装了 Tensorflow 和 Tensorflow-gpu (v.1.12.0)，因此它确实支持 CUDA。我还安装了 CUDA Toolkit v10 和 cuDNN 库，当我nvcc -V在命令提示符下运行时，我得到：

nvcc：NVIDIA (R) Cuda 编译器...

我正在使用 PyCharm，在 CPU 上运行 Keras 没有任何问题。但它不使用我的 GPU。

当我打字时

from keras import backend as K
K.tensorflow_backend._get_available_gpus()

它说

2018-11-25 10：47：57.448275：我tensorflow / core / platform / cpu_feature_gaurd.cc：141]您的CPU支持此TensorFlow二进制文件未编译使用的指令：AVX2

[ ]

我尝试过的：

1）我尝试卸载 Tensorflow 和 Tensorflow-gpu 并重新安装 Tensorflow-gpu，如上面的线程所述。不起作用，我的代码不再在 CPU 上运行，并给出了有关 Tensorflow 的错误。当我重新安装 Tensorflow 后，一切又正常了。

2）我尝试使用名为 Theano 的库，Keras 官方文档中提到了该库。当我添加以下行时

import theano
theano.config.device='gpu'
theano.config.floatX='float32'

它说

警告（theano.configdefaults）：如果使用 conda，则 g++ 不可用：'conda install m2w64-toolchain' C:\Users\HOME-PC\PyCharmProjects\env\lib\site-packages\theano\configdefaults.py.=:560 : UserWarning: DeprecationWarning: 没有 c++ 编译器。这已被弃用，并且在 theano 0.11 中将强制使用 c++ 编译器。warning.warn("DeprecationWarning: 没有 C++ 编译器。"

3）我尝试添加这些行，但也没有发生任何事情。

import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"]="0" 

我在这里缺少什么？我应该如何将 CUDA 引入 PyCharm？

Answer 1

问题出在 CUDA 的版本上。我已经安装了 CUDA v10.0，但 Tensorflow 似乎仅适用于 v9.0。安装了它，它就像一个魅力。