Tensorflow 呼び出し GPU の失敗の原因を見つける

問題の発見

画像を開いて Python と入力し、tensorflow をインポートしてプロンプト情報を確認すると、
ここに画像の説明を挿入します
ここで警告が表示されます。cudart64_101.dll が見つかりません

解決

インストールした cuda ディレクトリを見つけて、bin ファイルを入力し、cudart64_*.dll ファイルを見つけます。おそらくファイルは cudart64_102.dll ですが、このファイルを上記のように見つからないファイル名に変更するだけで済みます。
ここに画像の説明を挿入します

仕上げる

利用可能なデバイスを表示する

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

ここに画像の説明を挿入します
この時点で、成功したことを示す独自の GPU 情報が表示されます。

Supongo que te gusta

Origin blog.csdn.net/qq_43802454/article/details/121653611
Recomendado
Clasificación