CUDA11.3を選んだ理由は、Pytorchが2022/5/16時点では10.2と11.3に対応しているから。
インストール
https://developer.nvidia.com/cuda-toolkit-archive
↑ここからCUDA11.3を選択する
Installer Typeはdeb(network)を選んだほうが早くインストールできる、最後のsudo apt install cuda
はsudo apt install cuda-11-3
とする。こうした方が確実。
インストール後はbashrcを以下のとおりに変更する。
export PATH=/usr/local/cuda-11.3/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64:$LD_LIBRARY_PATH
バージョン確認
sudo apt install nvidia-cuda-toolkit
nvcc -V

CUDAを消したい or 入れ直したい
これで元どおり。
sudo apt-get --purge remove nvidia-*
sudo apt-get --purge remove cuda-*