CUDA Toolkit Linux 版のインストールメモ

2022年2月27日

はじめに

CUDA Tookit を Linux にインストールする。

バージョン

CUDA Toolkit 11.6

インストール環境

Ubuntu 20.04

ビデオカードを調べたい場合は以下のようにする。

$ lspci | grep -i nvidia

ちゃんとした情報が出てこない場合は、一度以下のように実行する。

$ sudo update-pciids

CUDA のインストール

こちら の指示に従う。選択肢が出てくるので、"Linux" - "x86_64" - "Ubuntu" - "20.04" と選択する。最後にインストールタイプを選択する必要があるが、ここでは "deb (local)" を選択する。コマンドの指示が出てくるので、実行する。

$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
$ sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
$ wget https://developer.download.nvidia.com/compute/cuda/11.6.0/local_installers/cuda-repo-ubuntu2004-11-6-local_11.6.0-510.39.01-1_amd64.deb
$ sudo dpkg -i cuda-repo-ubuntu2004-11-6-local_11.6.0-510.39.01-1_amd64.deb
$ sudo apt-key add /var/cuda-repo-ubuntu2004-11-6-local/7fa2af80.pub
$ sudo apt-get update
$ sudo apt-get -y install cuda

パスの設定を ~/.bashrc に追加。

export PATH=/usr/local/cuda-11.6/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64:$LD_LIBRARY_PATH

再起動する。

サンプルのコンパイル

サンプルをコンパイルする。その前に、必要なパッケージを入れておく。

$ sudo apt install mesa-common-dev libglu1-mesa-dev freeglut3-dev

コンパイル。

$ git clone https://github.com/NVIDIA/cuda-samples.git
$ cd cuda-samples
$ make

コンパイルできたら、試しに実行。

$ cd bin/x86_64/linux/release/
$ ./deviceQuery

ベンチマーク

nbody でベンチマークを行える。

$ ./nbody -benchmark -numbodies=100000

倍精度の場合はオプション "-fp64" を付ける。

ベンチマーク結果 (Quadro RTX 4000、FP32 7.1 TFLOPS、FP64 222.5 GFLOPS)