Google Compute EngineにNVIDIAドライバをインストールする方法

2024-03-07 20:31 Google Compute Engine NVIDIAドライバ インストール

  1. インスタンスの作成 まず、Google Cloud Consoleにアクセスし、Compute Engineのインスタンスを作成します。必要なスペック（CPU、メモリ、ディスク容量など）を選択し、インスタンスを作成します。

  2. NVIDIAドライバのインストールスクリプトのダウンロード 作成したインスタンスにSSHで接続し、以下のコマンドを実行して、NVIDIAドライバのインストールスクリプトをダウンロードします。

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
  1. ドライバのインストール 以下のコマンドを実行して、NVIDIAドライバをインストールします。
sudo apt-get -y install nvidia-driver-470
  1. ドライバのバージョン確認 以下のコマンドを実行して、正しくドライバがインストールされたかを確認します。
nvidia-smi

上記の手順に従うことで、Google Compute EngineのインスタンスにNVIDIAドライバを簡単にインストールすることができます。これにより、GPUを活用した高速な処理を実行することができます。

なお、上記の手順はUbuntu 20.04を前提としています。他のディストリビューションを使用している場合は、適宜コマンドを変更してください。また、ドライバのバージョンも最新のものに置き換えることができます。