プログラミング

cuDNNをインストールする

cuDNNはNVIDIA製品のGPUでTensorflowなどでディープラーニング、マシンラーニングのプログラムを動かすような場合に、CUDAやNVIDIAドライバーとともに必要となります。
以下に、Windows 10およびUbuntuへのcuDNNのインストール方法を示します。

cuDNNのダウンロードは以下より可能です。

NVIDIA cuDNN | NVIDIA Developer

ダウンロードにはログインおよびデベロッパープログラムへの参加が必要です。
少し手間がかかりますが、公式サイトからダウンロードをする場合にはこの手順はスキップできませんので、やるしかないようです。

過去のバージョンのcuDNNをダウンロード、インストールしたい場合、以下のアーカイブを利用します。
例えば、TensorFlowにCUDA 11.0が対応しており、11.0をインストールしたい、といった場合です。

cuDNN Archive | NVIDIA Developer

過去のバージョンのcuDNNのアーカイブ

Ubuntuの場合、上記手順でLinux・Ubuntuのパッケージをダウンロードし、.debファイル形式のパッケージファイルをダウンロードします。
これをsudo dpkg -i <library_name>.debでインストールし、インストールは完了します。

(参考)

Installation Guide :: NVIDIA Deep Learning cuDNN Documentation
nvidia – How can I install CuDNN on Ubuntu 16.04? – Ask Ubuntu

コメントを残す

メールアドレスが公開されることはありません。 * が付いている欄は必須項目です