Question

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

Keras with TensorFlow backend not using 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：I tensorflow / core / platform / cpu_feature_gaurd.cc：141]您的CPU支持该TensorFlow二进制文件未编译为使用的指令：AVX2

[]

我尝试过的事情：

1）我尝试卸载Tensorflow和Tensorflow-gpu并重新安装Tensorflow-gpu，如上述线程所述。没用，我的代码不再在CPU上运行，并给出了有关Tensorflow的错误。重新安装Tensorflow后，就可以了。

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

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

可能并非如此，但是安装和导入相同的库有时会造成混淆。更清楚地说，我的猜测是您的pycharm环境与默认的python环境不同，您在默认环境中安装了软件包并从pycharm的环境中导入了。为确保您在pycharm的环境中安装了软件包，

您可以尝试以下方法，从pycharm的python控制台中

！python -m pip install-升级tensorflow，keras

这将正确安装软件包，并且您可以确定安装没有错误

Answer 2

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

PyCharm中的Keras不使用GPU

2 个答案: