【发布时间】:2019-04-27 04:12:21
【问题描述】:
这些线程没有解决我的问题: 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): g++ 不可用,如果使用 conda: 'conda install m2w64-toolchain' C:\Users\HOME-PC\PyCharmProjects\env\lib\site-packages\theano\configdefaults.py.=:560: UserWarning: DeprecationWarning: there is no c++ compiler.This is deprecated and with theano 0.11 a c++ compiler will是强制性的。 warning.warn("DeprecationWarning: 没有 c++ 编译器。"
3) 我尝试添加这些行,但也没有任何反应。
import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"]="0"
我在这里缺少什么?我应该如何将 CUDA 引入 PyCharm?
【问题讨论】:
-
所以卸载 tensorflow 和 keras 没有帮助?
-
但是如果它无法加载本机运行时,那没关系,这就是我们想要的。本机运行时是 CPU 之一。
-
所以如果你没有 TF,只有 TF-gpu,你会遇到 keras 失败吗?您是否尝试过直接使用 TF keras 层而不是 keras?
-
他们在
tensorflow.keras,所以你应该可以像以前一样使用from TensorFlow import keras和使用API(tensorflow.org/guide/keras)。如果你使用这个API,你可以拥有tensorflow。 -
您确定您“完全”安装了 tensorflow 1.4 所需的 Cuda 版本吗? Tensoflow 很挑剔,它“完全”需要一个兼容的版本。我建议你去最新的 tensorflow 并安装教程中说的确切的 Cuda 版本。 -- 你不能同时拥有 tensorflow 和 tensorflow-gpu。仅安装 tensorflow GPU(如果您使用的是 conda,请尝试安装 keras-gpu)
标签: tensorflow keras pycharm