Ubuntu 20.04 での NVIDIA cuDNN のダウンロード

  1. NVIDIA cuDNN のウェブページを開く

    https://developer.nvidia.com/cuDNN

  2. ダウンロードしたいので,cuDNNのところにある「Download cuDNN」をクリック.
  3. メンバーシップに入る
  4. ログインする
  5. 調査の画面が出たときは,調査に応じる
  6. ライセンス条項の確認
  7. ダウンロードしたいバージョンを選ぶ

    ファイルを選ぶときの注意点:

    • 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.57.4.2 などの 7 系列を使う.8 系列は使わないこと.

    • 「for CUDA x.x」のところ: インストールした NVIDIA CUDA ツールキットのバージョン に合うものを選ぶこと.

    まず,「Archived cuDNN Releases」をクリック.

    先ほど CUDA 10.1 をインストールしたので, CUDA 10.1 に合うものを選ぶ.

  8. Linux 版が欲しいので Linux 版を選ぶ
  9. ダウンロードした .tgz ファイルを展開(解凍)する.

    次のように操作する.

    cd /usr/local
    sudo tar -xvzof <ダウンロードしたディレクトリ>/cudnn-10.1-linux-x64-v7.6.5.32.tgz 
    
  10. 展開の結果 cuda という名前のディレクトリができる.その中のサブディレクトリを確認しておく.
    ls -al /usr/local/cuda
    
  11. /usr/lib64 下にシンボリックリンク

    スーパーユーザで、次のコマンドを実行

    cd /usr/lib64
    for i in /usr/local/cuda/lib64/*; do echo $i; sudo ln -s $i .; done
    
  12. 「sudo /sbin/ldconfig」の実行
    sudo /sbin/ldconfig 
    

いまインストールした NVIDIA CUDA ツールキットのバージョンにあうものをインストールすること.

NVIDIA cuDNN のダウンロード URL: https://developer.nvidia.com/cudnn

ダウンロードするには,次の操作を行う.

  1. NVIDIA cuDNN のウェブページを開く

    https://developer.nvidia.com/cuDNN

  2. ダウンロードしたいので,cuDNNのところにある「Download cuDNN」をクリック.
  3. メンバーシップに入る
  4. ログインする
  5. ライセンス条項の確認
  6. CUDA のバージョン(例えば 11.0 )を確認のうえ, cuDNN のファイルを選ぶ.
  7. Linux 版を選ぶ.(cuDNN Library for Linux [x86_64] を選ぶ)
  8. ダウンロードした .tgz ファイルを展開(解凍)する.

インストールするには,次の操作を行う.

  1. 端末で,次のコマンドを実行する.
    cd /usr/local
    sudo tar -xvzof <ダウンロードしたディレクトリ>/cudnn-11.0-linux-x64-v8.0.5.39.tgz 
    
  2. /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 のインストールについて: 別ページ »で説明