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

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

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

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

前提条件:

  1. 最小20GBの空きディスク容量
  2. 推奨される2GBのRAM
  3. 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

CUDAバージョンを確認する

方法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

CUDAピンを/ etc / preferencesディレクトリに移動しますCUDAピンを/ etc / preferencesディレクトリに移動します

その後、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をダウンロード 無料のNVIDIADeveloper Programのメンバーになるには、登録する必要があります。

cuDNNlinuxのダウンロード

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をインストールする方法を学びました。 読んでくれてありがとう、あなたのフィードバックと提案を残してください。