cuDNN8が cuda-12.1 で動く?

先日の cuDNN の話の続き。

最新の cuDNN のダウンロードリンクを見ると、まだ CUDA-12.1 に対応していないように見えるのですが、実際には以下の手順で対応していることが分かります。

  • apt-get install cuda では 12.1 が入る
  • nvidia-smi を実行するとバージョンが 12.1 になっている。
  • apt-get update すると、12.0 のライブラリは autoremove 可能と言われる
  • 何より、remove しても cuDNN8 がちゃんと動く(Conv2d 等が機能する)

libcudnn8 の GPG キーがうまく機能しない問題は相変わらずなので、中途半端にインストールされている感も否めませんが、動いているのでよしとします。

投稿者について
みのしす

小さいときは科学者になろうとしたのに、その時にたまたま身に着けたプログラミングで未だに飯を食っているしがないおじさんです。(年齢的にはもうすぐおじいさん)

コメントを残す

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