1. ホーム
  2. クーダ

[解決済み】CuDNNのインストールを確認する方法は?

2022-04-11 07:09:45

質問

いろいろなところを探しましたが、インストールする方法ばかりで、インストールされているかどうかを確認する方法がわかりません。NVIDIAドライバがインストールされていることと、CUDAがインストールされていることは確認できましたが、CuDNNがインストールされていることを確認する方法がわかりません。助けていただけると大変助かります。

PS.

これはcaffeの実装のためです。現在、CuDNNを有効にしなくても全て動作しています。

どのように解決するのですか?

CuDNNのインストールは、CUDAディレクトリにファイルを配置するだけです。caffeのインストール時に、ルートとCuDNNオプションを正しく指定すれば、CuDNNでコンパイルされます。

を使用して確認することができます。 cmake . ディレクトリを作成する caffe/build を実行し cmake .. をクリックします。設定が正しければ、次のような行が表示されます。

-- Found cuDNN (include: /usr/local/cuda-7.0/include, library: /usr/local/cuda-7.0/lib64/libcudnn.so)

-- NVIDIA CUDA:
--   Target GPU(s)     :   Auto
--   GPU arch(s)       :   sm_30
--   cuDNN             :   Yes

すべて正しく動作していれば make の命令で、そこからcaffeをインストールします。