先日の 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 キーがうまく機能しない問題は相変わらずなので、中途半端にインストールされている感も否めませんが、動いているのでよしとします。
コメントを残す