216

私は多くの場所を検索しましたが、インストールされていることを確認する方法ではなく、インストールする方法だけが得られます。NVIDIA ドライバーがインストールされていること、および CUDA がインストールされていることは確認できますが、CuDNN がインストールされていることを確認する方法がわかりません。助けていただければ幸いです、ありがとう！

PS。
これはカフェの実装用です。現在、CuDNN を有効にせずにすべてが機能しています。

4

12 に答える 12

205

CuDNN のインストールは、いくつかのファイルをコピーするだけです。したがって、CuDNN がインストールされているかどうか (および使用しているバージョン) を確認するには、これらのファイルを確認するだけで済みます。

CuDNN をインストールする

ステップ 1: nvidia 開発者アカウントを登録し、ここから cudnn をダウンロードします(約 80 MB)。nvcc --versioncuda のバージョンを取得する必要があるかもしれません。

ステップ 2: cuda のインストール先を確認します。ほとんどの人にとって、それは/usr/local/cuda/. で確認できますwhich nvcc。

ステップ 3: ファイルをコピーします。

$ cd folder/extracted/contents
$ sudo cp include/cudnn.h /usr/local/cuda/include
$ sudo cp lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

バージョンを確認する

パスを調整する必要がある場合があります。インストールのステップ 2 を参照してください。

$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

編集：後のバージョンでは、これは次のようになる可能性があります（アリスへのクレジット）

$ cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

ノート

のようなエラーが出たとき

F tensorflow/stream_executor/cuda/cuda_dnn.cc:427] could not set cudnn filter descriptor: CUDNN_STATUS_BAD_PARAM

TensorFlow では、v5 の代わりに CuDNN v4 の使用を検討できます。

経由でインストールした Ubuntu ユーザーapt: https://askubuntu.com/a/767270/10425

于 2016-05-02T08:56:47.373 に答える
48

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 をインストールします。

于 2015-07-10T19:56:28.280 に答える
14

私は cuDNN 8.0 を使用していますが、上記の提案はどれもうまくいきませんでした。欲しい情報は にあった/usr/include/cudnn_version.hので、

cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

トリックをしました。

于 2020-08-24T20:19:10.387 に答える