インストール

https://developer.nvidia.com/cuda-toolkit-archive　からインストールしたいバージョンを探す。私の場合は cuda11.7, Ubuntu22.04 を選択した。

None

（必要があれば）libcudnn8 をインストール

JAX を使うときになぜか以下のようなエラーが出たため、追加で必要だった。

None

以下の一覧でlibcudnn8で検索をかけた。

https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/

Ubuntu20.04 を使っている場合は、上記リンクの"ubuntu2204"を"ubuntu2004"に変更すればいい。Ubuntu18.04 などの場合も同様。

先ほど cuda11.7 を入れたので、libcudnn88.5.0.96-1+cuda11.7amd64.deb を選ぶことにする。

None

以上がエラーなく終わったら、再起動する

sudo reboot

バージョン確認方法

Ubuntu のバージョンは以下コマンド

cat /etc/issue

x86_64 か arm かは以下コマンド

uname -m

完全にアンインストール

cuda toolkit を削除

sudo apt-get --purge remove "_cublas_" "cuda*" "nsight*"

Nvidia driver を削除

sudo apt-get --purge remove "_nvidia_"

ソースファイルがある場合（デフォルトでは/usr/local にある）、削除

sudo rm -rf /usr/local/cuda*

念の為、apt list も削除

sudo rm -rf /var/lib/apt/lists/*

不要なパッケージを削除

sudo apt autoremove