tensorflowでcublasLt64_12.dllが見つからないと言われ、gpuが使えない

Question

実現したいこと

TensorflowでGPUを使いたいが、エラーが出て動かない。

前提

以前はレンタルサーバー(Ubuntu)で機械学習を行っていましたが、手元のWindowsマシンに移行中です。

Python, TensorFlow, CUDA, cuDNNのバージョンを確認した上でインストールしました。各バージョンは補足に記載してあります。

以下のコードで、GPUが認識されていることは確認済みです。

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

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 1720156617681741145
xla_global_id: -1
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 9383706624
locality {
  bus_id: 1
  links {
  }
}

発生している問題・エラーメッセージ

学習を行おうとすると、以下のエラーが出ました。

Could not locate cublasLt64_12.dll. Please make sure it is in your library path!

cublastLt64_12.dllが見つからないというエラーですが、ググっても情報は見つかりませんでした。
CUDAのバージョンは11.2なのに、なぜ64_12.dllが要求されるかわかりません。
なお、最初に誤ってCUDA v12.1をインストールしています(コントロールパネルから消去済み)。

試したこと

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\bin内の 'cublasLt64_11.dll', 'cublas64_11.dll' を、それぞれ'cublasLt64_12.dll', 'cublas64_12.dll'にリネームした。結果、エラーは出ないがCPUで学習が始まりました。

補足情報（FW/ツールのバージョンなど）

Windows10
Python:3.7
CUDA:11.2
cuDNN:8.1.3
Tensorflow-gpu:2.10

Comments

[jbpb0]

「補足情報（FW/ツールのバージョンなど）」に

> CUDA:12.1

とありますが、

> CUDAのバージョンは11.2なのに、

> 最初に誤ってCUDA v12.1をインストールしています(コントロールパネルから消去済み)。

ならば、実際は「cuda 11.2」でしょうか？

もしそうなら、質問を編集して、上記を直してください

[jbpb0]

「C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA」には、名前が「v12」で始まるディレクトリはありませんでしょうか？

[otatatatat]

失礼しました。仰る通りです。修正します。

[otatatatat]

はい、ディレクトリはv11.2のみで、v12のものはありません。

[jbpb0]

https://developer.nvidia.com/rdp/cudnn-archive
を見ても

> cuDNN:8.1.3

は見当たらないのですが、どれを入れたのでしょうか？

また、間違えて(cuda 11.2に)cuda 12用のcudnnを入れてる、ということはありませんでしょうか？

[otatatatat]

ありがとうございます。
cuDNNを改めて確認したところ、v 8.8.1がインストールされていました。
8.1系をインストールしたところ、無事に動きました。
初歩的な確認ミスでお手数おかけしてすみません。ありがとうございました！

Answer 1

cuDNNのバージョンが間違っていました。C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2以下を正しいバージョンのもので書き換えたところ、動作しました。