CUDA11.3 インストール(Ubuntu18.04)

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-*

参考

• https://qiita.com/1O1/items/c7b0cec503d48b2efde1
• https://qiita.com/Yoshitaka_Engineer/items/bd8edcb36396df878641