Ubuntu20.04にCUDAをインストールする方法

CUDA は、並列コンピューティングプラットフォームであり、日常の汎用コンピューティングにGPUを活用する際に優れたユーザーエクスペリエンスを提供するプログラミングモデルです。 CUDAはCの略です ompute U nified D evice A rchitecture。 NVIDIAによって作成されています。

CUDAは、CUDAツールキット（コンパイラー、プロファイル、およびデバッガー）、ソフトウェアドライバー、およびCUDASDKで構成されます。ソフトウェア層は、GPUの仮想命令セットと並列計算要素への直接アクセスを提供します。ディープラーニングの研究とフレームワークの開発には、 cuDNNを使用します高性能GPUアクセラレーション用。これは、ディープニューラルネットワーク用のプリミティブのライブラリです。

このガイドでは、Ubuntu20.04にCUDAとCuDNNをインストールする方法を学びます。。これを行うには2つの方法があります。 CUDAはUbuntuリポジトリからインストールできます（これが最も簡単な方法です）。または、最新バージョンのCUDAを提供するCUDAリポジトリからインストールできます。

前提条件：

最低20GBの空きディスク容量
2GBのRAMを推奨
NVIDIAグラフィックスドライバーがインストールされている必要があります。 CUDAに付属しているドライバーは最新バージョンではない可能性があります。

方法1：UbuntuリポジトリからCUDAをインストールする

まず、Ubuntu20.04インスタンスのパッケージリストを更新します。

$ sudo apt update

次に、次のようにAPTパッケージマネージャーを使用してCUDAツールキットをインストールします。インストールは集中的に行われるため、高速で安定したインターネット接続と最低10Gのディスク容量を確保することに注意してください。

$ sudo apt install nvidia-cuda-toolkit

インストールが完了したら、CUDAの場所をシェルに通知するために、PATHにCUDAを追加する必要があります。これを行うには、.bashrcファイルでPATHを指定します。したがって、コマンドラインテキストエディタを使用してファイルを開きます。

$ sudo vim ~/.bashrc

ファイルの最後に、次のコード行をコピーして貼り付けます。

# set PATH for cuda 10.1 installation

if [ -d "/usr/local/cuda-10.1/bin/" ]; then

export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

fi

次に、.bashrcファイルの変更をリロードします。

$ source ~/.bashrc

次に、次のコマンドを使用して、インストールされているCUDAのバージョンを確認します。

$ nvcc --version

方法2：CUDAリポジトリからCUDAをインストールする

前述のように、CUDAリポジトリは最新バージョンのCUDAを提供します。したがって、最新バージョンをインストールする場合は、ここに概説されている手順に従ってください。

まず、CUDAリポジトリピンをダウンロードします：

$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin

次に、ピンを/etc/preferences.d ディレクトリを作成し、名前をcuda-repository-pin-600 。

$ sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600

その後、CUDADebianファイルをダウンロードします。このガイドを書いている時点では、ファイルが非常に大きいことに注意してください。正確には2.3Gです。良好なインターネット接続と十分なディスク容量があることを確認してください。

$ wget https://developer.download.nvidia.com/compute/cuda/11.4.2/local_installers/cuda-repo-ubuntu2004-11-4-local_11.4.2-470.57.02-1_amd64.deb

ダウンロードしたら、dpkg ファイルを実行するユーティリティ。

sudo dpkg -i cuda-repo-ubuntu2004-11-4-local_11.4.2-470.57.02-1_amd64.deb

次に、次のようにCUDAGPGキーをインストールします。

$ sudo apt-key add /var/cuda-repo-ubuntu2004-11-4-local/7fa2af80.pub

最後に、パッケージリストを更新し、APTパッケージマネージャーを使用してCUDAをインストールします。

$ sudo apt update

$ sudo apt install cuda -y

CuDNNをインストールする

CuDNNにはCudaは付属していません。 CuDNNをダウンロードするには、無料のNVIDIADeveloperProgramのメンバーになるために登録する必要があります。

NVIDIA Developer Programに登録した後、cuDNNの最新バージョンをダウンロードできます。ここでは、Linux用のcuDNNライブラリをダウンロードしています。

まず、cuDNNパッケージを抽出します：

$ tar -xzvf cudnn-10.2-linux-x64-v8.2.4.15.tgz

次に、次のファイルをCUDAディレクトリにコピーします。

$ sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

結論

このチュートリアルでは、Ubuntu20.04にCUDAとcuDNNをインストールする方法を学びました。読んでいただきありがとうございます。フィードバックや提案を残してください。