Ubuntu 20.04 での NVIDIA cuDNN のダウンロード
- NVIDIA cuDNN のウェブページを開く
- ダウンロードしたいので,cuDNNのところにある「Download cuDNN」をクリック.
- メンバーシップに入る
- ログインする
- 調査の画面が出たときは,調査に応じる
- ライセンス条項の確認
- ダウンロードしたいバージョンを選ぶ
ファイルを選ぶときの注意点:
- TensorFlow 2.4 の GPU 版での,cuDNN のバージョンは 7(現時点で,NVIDIA CUDA ツールキット 11.0 で動く最新版).
TensorFlow 2.3, 2.2, 2.1 の GPU 版での,cuDNN のバージョンは 7.6.そして,TensorFlow 1.13 以上 TensorFlow 2.0 までの GPU 版での,cuDNN のバージョンは7.6.5 や 7.4.2 などの 7 系列を使う.8 系列は使わないこと.
- 「for CUDA x.x」のところ: インストールした NVIDIA CUDA ツールキットのバージョン に合うものを選ぶこと.
まず,「Archived cuDNN Releases」をクリック.
先ほど CUDA 10.1 をインストールしたので, CUDA 10.1 に合うものを選ぶ.
- TensorFlow 2.4 の GPU 版での,cuDNN のバージョンは 7(現時点で,NVIDIA CUDA ツールキット 11.0 で動く最新版).
- Linux 版が欲しいので Linux 版を選ぶ
- ダウンロードした .tgz ファイルを展開(解凍)する.
次のように操作する.
cd /usr/local sudo tar -xvzof <ダウンロードしたディレクトリ>/cudnn-10.1-linux-x64-v7.6.5.32.tgz
- 展開の結果 cuda という名前のディレクトリができる.その中のサブディレクトリを確認しておく.
ls -al /usr/local/cuda
- /usr/lib64 下にシンボリックリンク
スーパーユーザで、次のコマンドを実行
cd /usr/lib64 for i in /usr/local/cuda/lib64/*; do echo $i; sudo ln -s $i .; done
- 「sudo /sbin/ldconfig」の実行
sudo /sbin/ldconfig
いまインストールした NVIDIA CUDA ツールキットのバージョンにあうものをインストールすること.
NVIDIA cuDNN のダウンロード URL: https://developer.nvidia.com/cudnn
ダウンロードするには,次の操作を行う.
- NVIDIA cuDNN のウェブページを開く
- ダウンロードしたいので,cuDNNのところにある「Download cuDNN」をクリック.
- メンバーシップに入る
- ログインする
- ライセンス条項の確認
- CUDA のバージョン(例えば 11.0 )を確認のうえ, cuDNN のファイルを選ぶ.
- Linux 版を選ぶ.(cuDNN Library for Linux [x86_64] を選ぶ)
- ダウンロードした .tgz ファイルを展開(解凍)する.
インストールするには,次の操作を行う.
- 端末で,次のコマンドを実行する.
cd /usr/local sudo tar -xvzof <ダウンロードしたディレクトリ>/cudnn-11.0-linux-x64-v8.0.5.39.tgz
- /usr/lib/x86_64-linux-gnu/ 下にシンボリックリンク
端末で,次のコマンドを実行する.
cd /usr/lib/x86_64-linux-gnu/ for i in /usr/local/cuda/lib64/*; do echo $i; sudo rm -f `basename $i`; sudo ln -s $i .; done
詳しくは,NVIDIA cuDNN のインストールについて: 別ページ »で説明