PyCharm中的Keras未使用GPU



这些线程没有解决我的问题:Keras在Pycharm上不使用GPU,Pycharm具有python 3.5和Tensorflow 1.4

带有TensorFlow后端的Keras不使用GPU

我在我的电脑上安装了Tensorflow和Tensorflow gpu(v.1.12.0(,它运行的是Windows 10,并且有GTX 750 Ti显卡,所以它确实支持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官方文档中提到了这个库。当我添加以下行时

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

上面写着

警告(theano.configdefaults(:如果使用conda:'conda-install m2w64 toolchain',g++不可用C: \Users\HOME-PC\PyCharmProjects\env\lib\site packages\theano\configdefaults.py.=:560:UserWarning:弃用警告:没有C++编译器。这是不推荐使用的,对于0.11版本,c++编译器将是强制性的。warning.warn("弃用警告:没有c++编译器。">

3(我试着添加这些行,但什么也没发生。

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

我在这里错过了什么?我应该如何将CUDA介绍给PyCharm?

问题出在CUDA的版本上。我安装了CUDA v10.0,但Tensorflow似乎只能与v9.0配合使用。安装它,它就像一个魅力。

情况可能并非如此,但安装和导入同一个库有时会令人困惑。更清楚地说,我猜你的pycharm环境与默认的python环境不同,你在默认环境中安装软件包,并从pycharm的环境中导入。为了确保您在pycharm的环境中安装了一个软件包,

您可以尝试以下操作,从pycharm的python控制台

python-m pip安装-升级tensorflow、keras

这将正确安装软件包,并且你可以确信安装没有出现故障

最新更新