在訓練神經網絡時，Keras不使用我的Nvidia GPU

Question

Keras / TensorFlow不使用我的GPU。

為了嘗試讓我的GPU使用tensorflow，我嘗試安裝tensorflow-gpu（我在Windows上使用Python 3.6.8）：

pip3 install tensorflow-gpu --user

python -m notebook

import tensorflow as tf

然后我得到以下錯誤：

ImportError ... Traceback (most recent call last),

ImportError: DLL load failed: Le module spécifié est introuvable.

ImportError ... Traceback (most recent call last)

然后我做pip3 install tensorflow, python - notebook ，然后import tensorflow as tf工作，但當我繼續：

from tensorflow.python.client import device_lib

print(device_lib.list_local_devices())

print('Tensorflow: ', tf._ _ version _ _)

我得到：

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 587921620497715868
]
Tensorflow:  1.13.1

這意味着沒有找到GPU（我的聯想Thinkpad P72上有一個Intel UHD Graphics P630和一個Nvidia Quadro P5200）。

另外，請注意，當我這樣做時！nvidia-smi我看到Nvidia被檢測到（作為GPU [0]），所以我不明白為什么Tensorflow / Keras在訓練神經網絡時不使用它。

謝謝你的幫助。

Answer 1

我想我解決了這個問題:)

安裝Anaconda3之后，我在Anaconda powershell中繼續這樣做：

conda create -n tensorflowgpuproject python = 3.5 tensorflow-gpu = 1.10.0 ipython = 6.5.0 keras matplotlib jupyter scikit-learn numpy = 1.14.5 setuptools = 39.1.0 prompt_toolkit = 1.0.15

conda激活tensorflowgpuproject

python -m pip install --upgrade pip --user

setx路徑“％path％; C：\\ Users \\ n-bou \\ AppData \\ Roaming \\ Python \\ Python35 \\ Scripts”

pip uninstall prompt_toolkit（最后一個命令是必要的，因為兩個版本的prompt_toolkit之間存在沖突）

然后我能夠在筆記本開頭使用常用的“import tensorflow as tf”（調用tensorflow-gpu）執行包含UNet代碼的python筆記本。