インストール

cuDNNをインストールする

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

Shou Arisaka
1 分で読める
2022年1月19日

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

Image

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

この記事をシェア

Shou Arisaka 2022年1月19日

🔗 リンクをコピー