<p>＊これは<a href="https://adventar.org/calendars/5690" target="_blank" rel="nofollow noopener noreferrer">KCS AdventCalendar2020</a> 8日目の記事です＊</p>
<p><a href="https://kcs1959.jp/archives/6077/general/%E5%AE%9F%E9%9A%9B%E3%81%AB%E3%82%B5%E3%83%B3%E3%83%88%E3%83%A9%E3%82%92%E4%BD%9C%E6%9B%B2%E3%81%97%E3%81%A6%E3%81%BF%E3%81%9F" target="_blank" rel="nofollow noopener noreferrer">←7日目</a> | <a href="https://kcs1959.jp/archives/6064/general/p5-js%E5%A4%A7%E5%A5%BD%E3%81%8D-%E6%A5%BD%E3%81%A0%E3%81%8B%E3%82%89" target="_blank" rel="nofollow noopener noreferrer">9日目→</a></p>
<hr>
<h2 id="%E3%81%AF%E3%81%98%E3%82%81%E3%81%AB">
<a class="header-anchor-link" href="#%E3%81%AF%E3%81%98%E3%82%81%E3%81%AB" aria-hidden="true"></a> はじめに</h2>
<p>NVIDIAのGPUを使用する際，トラブルに見舞われる方の多くはドライバのインストールでコケていらっしゃると思います．<br>
また，NVIDIAのGPUドライバにはいくつかバリエーションがあり，どれをインストールしたらよいか分からない，という方もいらっしゃると思います．<br>
この記事がそのような方の一助となれば幸いです．</p>
<p>筆者の環境はUbuntu 18.04 LTS (Linux Kernel 4.15.0 および 5.4.0)です．ご自身の環境によって適宜読み替えてください．</p>
<p>なお，<a href="#%E3%83%89%E3%83%A9%E3%82%A4%E3%83%90%E3%81%AE%E3%83%90%E3%83%BC%E3%82%B8%E3%83%A7%E3%83%B3%E3%82%92%E5%9B%BA%E5%AE%9A%E3%81%97%E3%81%9F%E3%81%84%E5%A0%B4%E5%90%88">バージョン指定する方法</a>を使用する場合のUbuntu用スクリプトを<a href="https://gist.github.com/190ikp/b368905ebc7e748d065e5f57d59892df" target="_blank" rel="nofollow noopener noreferrer">gist</a>に置いてあります．こちらのスクリプトでは，</p>
<ul>
<li><code>nvidia-driver-"$NVIDIA_DRIVER_VERSION"</code></li>
<li><code>cuda-toolkit-"$CUDA_VERSION"</code></li>
<li><code>nvidia-container-toolkit</code></li>
</ul>
<p>のインストールを行うことができます．ご自由にお使いください．</p>
<h2 id="%E3%82%A4%E3%83%B3%E3%82%B9%E3%83%88%E3%83%BC%E3%83%AB%E6%96%B9%E6%B3%95">
<a class="header-anchor-link" href="#%E3%82%A4%E3%83%B3%E3%82%B9%E3%83%88%E3%83%BC%E3%83%AB%E6%96%B9%E6%B3%95" aria-hidden="true"></a> インストール方法</h2>
<p>先にリポジトリ等の準備をしておきます．これは，これから紹介するどの方法においても共通です．</p>
<p>まず，GPUドライバを配布しているNVIDIAのリポジトリをパッケージマネージャに登録します．<br>
CPUがx86系(Intel or AMD)でOSがUbuntu 18.04の場合は<br>
<a href="https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64" target="_blank" rel="nofollow noopener noreferrer">https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64</a><br>
を登録します．この際にGPG公開鍵も登録しておきましょう．</p>
<div class="code-block-container"><pre class="language-shell"><code class="language-shell">$ <span class="token function">curl</span> <span class="token parameter variable">-fsSL</span> https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub <span class="token operator">|</span>
    <span class="token function">sudo</span> gpg --no-default-keyring <span class="token parameter variable">--keyring</span> /usr/share/keyrings/nvidia-cuda.gpg <span class="token parameter variable">--import</span>
$ <span class="token builtin class-name">echo</span> <span class="token string">"deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 /"</span> <span class="token operator">|</span>
    <span class="token function">sudo</span> <span class="token function">tee</span> /etc/apt/sources.list.d/cuda.list
</code></pre></div><details><summary>リポジトリ登録時の注意点</summary><div class="details-content">
<aside class="msg alert"><span class="msg-symbol">!</span><div class="msg-content">
<p><code>apt-key</code>コマンドは非推奨になりました．ここでは<code>/usr/share/keyrings/</code>に鍵を登録するようにしています．</p>
</div></aside>
<aside class="msg alert"><span class="msg-symbol">!</span><div class="msg-content">
<p><code>cuda</code>リポジトリのGPG公開鍵は2022年4月に更新されました．こちらに記載している鍵は2023年10月時点のものです．</p>
</div></aside>
</div></details>
<p>対応するディストリビューションの一覧は<br>
<a href="https://developer.download.nvidia.com/compute/cuda/repos/" target="_blank" rel="nofollow noopener noreferrer">https://developer.download.nvidia.com/compute/cuda/repos/</a><br>
から探してください．</p>
<details><summary>機械学習等で利用する場合</summary><div class="details-content">
<p>機械学習等で利用するのであれば，CuDNNのリポジトリも登録しておくと便利です．<br>
対応ディストリビューションの一覧は<br>
<a href="https://developer.download.nvidia.com/compute/machine-learning/repos/" target="_blank" rel="nofollow noopener noreferrer">https://developer.download.nvidia.com/compute/machine-learning/repos/</a><br>
から探してください．<br>
以下にリポジトリの登録例を示しておきます．</p>
<div class="code-block-container"><pre class="language-shell"><code class="language-shell"><span class="token function">curl</span> <span class="token parameter variable">-fsSL</span> <span class="token string">"https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub"</span> <span class="token operator">|</span>
    <span class="token function">sudo</span> gpg --no-default-keyring <span class="token parameter variable">--keyring</span> /usr/share/keyrings/nvidia-ml.gpg <span class="token parameter variable">--import</span>
<span class="token builtin class-name">echo</span> <span class="token string">"deb [signed-by=/usr/share/keyrings/nvidia-ml.gpg] https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /"</span> <span class="token operator">|</span>
    <span class="token function">sudo</span> <span class="token function">tee</span> /etc/apt/sources.list.d/nvidia-ml.list
</code></pre></div>
</div></details>
<p>次にカーネルのヘッダファイルとビルドツールをインストールします．</p>
<div class="code-block-container"><pre class="language-shell"><code class="language-shell">$ <span class="token function">sudo</span> <span class="token function">apt</span> update
$ <span class="token function">sudo</span> <span class="token function">apt</span> upgrade <span class="token parameter variable">-y</span>
$ <span class="token function">sudo</span> <span class="token function">apt</span> <span class="token function">install</span> <span class="token parameter variable">-y</span> <span class="token punctuation">\</span>
    <span class="token string">"linux-headers-<span class="token variable"><span class="token variable">$(</span><span class="token function">uname</span> <span class="token parameter variable">-r</span><span class="token variable">)</span></span>"</span> <span class="token punctuation">\</span>
    build-essential
</code></pre></div><p>これで準備完了です．</p>
<h3 id="%E6%9C%80%E3%82%82%E3%82%B7%E3%83%B3%E3%83%97%E3%83%AB%E3%81%AA%E3%83%91%E3%82%BF%E3%83%BC%E3%83%B3">
<a class="header-anchor-link" href="#%E6%9C%80%E3%82%82%E3%82%B7%E3%83%B3%E3%83%97%E3%83%AB%E3%81%AA%E3%83%91%E3%82%BF%E3%83%BC%E3%83%B3" aria-hidden="true"></a> 最もシンプルなパターン</h3>
<p>ほとんどの方はこちらの方法で十分です．<br>
この方法でインストールしたドライバは，パッケージのアップデートに伴ってバージョンが変更されます．ドライバのバージョンを固定したい場合は次の<a href="#%E3%83%89%E3%83%A9%E3%82%A4%E3%83%90%E3%81%AE%E3%83%90%E3%83%BC%E3%82%B8%E3%83%A7%E3%83%B3%E3%82%92%E5%9B%BA%E5%AE%9A%E3%81%97%E3%81%9F%E3%81%84%E5%A0%B4%E5%90%88">ドライバのバージョンを固定したい場合</a>を参照してください．</p>
<p>やり方は非常にシンプルで，<code>cuda-drivers</code>をインストールするだけです．</p>
<div class="code-block-container"><pre class="language-shell"><code class="language-shell">$ <span class="token function">sudo</span> <span class="token function">apt</span> <span class="token function">install</span> <span class="token parameter variable">-y</span> cuda-drivers
</code></pre></div><p>インストールができたらマシンを再起動しましょう．</p>
<aside class="msg alert"><span class="msg-symbol">!</span><div class="msg-content">
<p>パッケージインストール後の再起動はどの方法でも必要です．</p>
</div></aside>
<p><code>nvidia-smi</code>を叩いて以下のような出力が得られればOKです．</p>
<div class="code-block-container"><pre><code>$ nvidia-smi
Tue Dec  8 01:08:31 2020       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 450.80.02    Driver Version: 450.80.02    CUDA Version: 11.0     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  GeForce RTX 208...  Off  | 00000000:3B:00.0 Off |                  N/A |
|  0%   37C    P0    62W / 300W |      0MiB / 11019MiB |      1%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
|   1  GeForce RTX 208...  Off  | 00000000:AF:00.0 Off |                  N/A |
| 32%   38C    P0    25W / 300W |      0MiB / 11019MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|  No running processes found                                                 |
+-----------------------------------------------------------------------------+
</code></pre></div><p><code>cuda-drivers</code>をインストールすると，</p>
<ul>
<li>最新のGPUドライバ</li>
<li>X11 / Wayland用のパッケージ<br>
(よくわからない方は「GUIに関わるパッケージ」と考えていただければ十分です)</li>
<li>各種管理用ユーティリティ</li>
</ul>
<p>等がインストールされます．詳しく知りたい方は依存関係を調べてみてください．<br>
また，よくトラブルの原因になるnouveauについては，自動的にブラックリストに入れられるため特に操作を行う必要はありません．</p>
<p>この後CUDAを使用する場合は，<code>cuda-toolkit-&lt;version&gt;</code>パッケージをインストールします．<br>
<code>cuda-&lt;version&gt;</code>と異なり，このパッケージにはGPUドライバが含まれていません．そのためデバイスドライバとCUDAのバージョンを別々に管理できるようになり，非常に便利です．</p>
<p>また，<code>cuda-drivers</code> + <code>nvidia-container-toolkit</code>をホストにインストールしてDocker / LXDコンテナ内からGPUを使用する場合，<code>cuda-&lt;version&gt;</code>をコンテナ内環境にインストールすると依存関係が壊されます．<br>
これは<code>cuda-&lt;version&gt;</code>パッケージに含まれるGPUドライバとホストのドライバが干渉するためです．<br>
そのため，この場合もコンテナ内には<code>cuda-toolkit-&lt;version&gt;</code>をインストールするようにしてください．</p>
<h3 id="%E3%83%89%E3%83%A9%E3%82%A4%E3%83%90%E3%81%AE%E3%83%90%E3%83%BC%E3%82%B8%E3%83%A7%E3%83%B3%E3%82%92%E5%9B%BA%E5%AE%9A%E3%81%97%E3%81%9F%E3%81%84%E5%A0%B4%E5%90%88">
<a class="header-anchor-link" href="#%E3%83%89%E3%83%A9%E3%82%A4%E3%83%90%E3%81%AE%E3%83%90%E3%83%BC%E3%82%B8%E3%83%A7%E3%83%B3%E3%82%92%E5%9B%BA%E5%AE%9A%E3%81%97%E3%81%9F%E3%81%84%E5%A0%B4%E5%90%88" aria-hidden="true"></a> ドライバのバージョンを固定したい場合</h3>
<p>この場合は，<code>nvidia-driver-&lt;version&gt;</code>をインストールします．<br>
こちらを利用した場合は，ドライバのバージョンが<code>&lt;version&gt;</code>で指定したものに固定されます．<br>
インストール可能なバージョンは<code>apt search</code>で調べることができます．</p>
<div class="code-block-container"><pre class="language-shell"><code class="language-shell">$ <span class="token function">apt</span> search ^nvidia-driver
Sorting<span class="token punctuation">..</span>. Done
Full Text Search<span class="token punctuation">..</span>. Done
nvidia-driver-390/bionic-updates,bionic-security <span class="token number">390.138</span>-0ubuntu0.18.04.1 amd64
  NVIDIA driver metapackage

nvidia-driver-410/unknown <span class="token number">410.129</span>-0ubuntu1 amd64
  NVIDIA driver metapackage

nvidia-driver-418/bionic-updates <span class="token number">430.50</span>-0ubuntu0.18.04.2 amd64
  Transitional package <span class="token keyword">for</span> nvidia-driver-430

nvidia-driver-418-server/bionic-updates <span class="token number">418.152</span>.00-0ubuntu0.18.04.1 amd64
  NVIDIA Server Driver metapackage

nvidia-driver-430/unknown,unknown <span class="token number">455.45</span>.01-0ubuntu1 amd64
  Transitional package <span class="token keyword">for</span> nvidia-driver-455

nvidia-driver-435/bionic-updates <span class="token number">455.38</span>-0ubuntu0.18.04.1 amd64
  Transitional package <span class="token keyword">for</span> nvidia-driver-455

nvidia-driver-440/bionic-updates <span class="token number">450.80</span>.02-0ubuntu0.18.04.2 amd64
  Transitional package <span class="token keyword">for</span> nvidia-driver-450

nvidia-driver-440-server/bionic-updates <span class="token number">440.95</span>.01-0ubuntu0.18.04.1 amd64
  NVIDIA Server Driver metapackage

nvidia-driver-450/unknown <span class="token number">450.80</span>.02-0ubuntu1 amd64
  NVIDIA driver metapackage

nvidia-driver-450-server/bionic-updates <span class="token number">450.80</span>.02-0ubuntu0.18.04.3 amd64
  NVIDIA Server Driver metapackage

nvidia-driver-455/unknown,unknown <span class="token number">455.45</span>.01-0ubuntu1 amd64
  NVIDIA driver metapackage
</code></pre></div><p>2020年12月現在の最新のバージョンは<code>nvidia-driver-455</code>です．<br>
なお，前項で<code>cuda-drivers</code>をインストールしましたが，これはその時点で最新の<code>nvidia-driver</code>をインストールするような挙動になります．つまり2020年12月時点では<code>nvidia-driver-455</code>がインストールされます．</p>
<p>また，<code>nvidia-driver-&lt;version&gt;-server</code>というパッケージが見えていますが，こちらは見ての通りbionic-updatesで提供されています．また，依存する各パッケージも<code>server</code>プレフィックスのついたものとなっています．必要に応じて選択してください．</p>
<h3 id="gui%E3%81%8C%E4%B8%8D%E8%A6%81%E3%81%AA%E5%A0%B4%E5%90%88">
<a class="header-anchor-link" href="#gui%E3%81%8C%E4%B8%8D%E8%A6%81%E3%81%AA%E5%A0%B4%E5%90%88" aria-hidden="true"></a> GUIが不要な場合</h3>
<p>純粋にGPGPU用途でGPUを使用する場合，<code>headless</code>プレフィックスのついたパッケージを使用できます．<br>
この場合は<code>nvidia-headless-&lt;version&gt;</code>をインストールしてください．<br>
こちらも<code>nvidia-headless-&lt;version&gt;-server</code>という，bionic-updatesで提供されるバージョンが選択できます．</p>
<p>なお，このパッケージには管理用のユーティリティも含まれません．<br>
<code>nvidia-smi</code>等が必要な場合は<code>nvidia-utils-&lt;version&gt;</code> (<code>nvidia-utils-&lt;version&gt;-server</code>)をインストールしてください．</p>
<p>また，<code>headless</code>にはOpenGLやVulkan等の主要なグラフィックスライブラリも含まれません．<br>
レンダリングファーム等で使用する場合は前項の<code>nvidia-driver</code>か<code>cuda-drivers</code>をインストールされることをおすすめします．</p>
<h2 id="%E3%81%BE%E3%81%A8%E3%82%81">
<a class="header-anchor-link" href="#%E3%81%BE%E3%81%A8%E3%82%81" aria-hidden="true"></a> まとめ</h2>
<ul>
<li>右も左もわからない場合は<code>cuda-drivers</code>を入れておきましょう．幸せになれます．</li>
<li>特に支障がなければ<code>cuda-drivers</code>を入れておきましょう．楽です．</li>
<li>サーバ用途であれば<code>nvidia-driver-&lt;version&gt;-server</code>を入れておきましょう．<br>
幸せになれるかもしれません．</li>
<li>Tesla欲しい．</li>
</ul>
<p>以上です．</p>
<hr>
<p><a href="https://kcs1959.jp/archives/6077/general/%E5%AE%9F%E9%9A%9B%E3%81%AB%E3%82%B5%E3%83%B3%E3%83%88%E3%83%A9%E3%82%92%E4%BD%9C%E6%9B%B2%E3%81%97%E3%81%A6%E3%81%BF%E3%81%9F" target="_blank" rel="nofollow noopener noreferrer">←7日目</a> | <a href="https://kcs1959.jp/archives/6064/general/p5-js%E5%A4%A7%E5%A5%BD%E3%81%8D-%E6%A5%BD%E3%81%A0%E3%81%8B%E3%82%89" target="_blank" rel="nofollow noopener noreferrer">9日目→</a></p>


NVIDIAのGPUドライバを楽にインストールしたい

ドライバのバージョンを固定したい場合

tshino118

190ikp

<h2 id="%E3%83%AA%E3%83%9D%E3%82%B8%E3%83%88%E3%83%AA%E3%82%AD%E3%83%BC%E3%81%AE%E6%9B%B4%E6%96%B0%E3%81%8C%E3%81%82%E3%82%8A%E3%81%BE%E3%81%97%E3%81%9F(apr-28%2C-2022)%E3%80%82">
<a class="header-anchor-link" href="#%E3%83%AA%E3%83%9D%E3%82%B8%E3%83%88%E3%83%AA%E3%82%AD%E3%83%BC%E3%81%AE%E6%9B%B4%E6%96%B0%E3%81%8C%E3%81%82%E3%82%8A%E3%81%BE%E3%81%97%E3%81%9F(apr-28%2C-2022)%E3%80%82" aria-hidden="true"></a> リポジトリキーの更新がありました(Apr 28, 2022)。</h2>
<p><span class="embed-block zenn-embedded zenn-embedded-card"><iframe id="zenn-embedded__eb14f5102f993" src="https://embed.zenn.studio/card#zenn-embedded__eb14f5102f993" data-content="https%3A%2F%2Fdeveloper.nvidia.com%2Fblog%2Fupdating-the-cuda-linux-gpg-repository-key%2F" frameborder="0" scrolling="no" loading="lazy"></iframe></span><a href="https://developer.nvidia.com/blog/updating-the-cuda-linux-gpg-repository-key/" style="display:none" target="_blank" rel="nofollow noopener noreferrer">https://developer.nvidia.com/blog/updating-the-cuda-linux-gpg-repository-key/</a></p>
<h2 id="%E3%82%A4%E3%83%B3%E3%82%B9%E3%83%88%E3%83%BC%E3%83%AB%E6%96%B9%E6%B3%95-(distro%3Dubuntu1804%2C-arch%3Dx86_64%E3%81%AE%E5%A0%B4%E5%90%88)">
<a class="header-anchor-link" href="#%E3%82%A4%E3%83%B3%E3%82%B9%E3%83%88%E3%83%BC%E3%83%AB%E6%96%B9%E6%B3%95-(distro%3Dubuntu1804%2C-arch%3Dx86_64%E3%81%AE%E5%A0%B4%E5%90%88)" aria-hidden="true"></a> インストール方法 (distro=ubuntu1804, arch=x86_64の場合)</h2>
<p>###既にキーが登録されている場合<br>
古いキーを削除</p>
<div class="code-block-container"><pre><code>sudo apt-key del 7fa2af80
</code></pre></div><p>###キーの登録<br>
変数に代入</p>
<div class="code-block-container"><pre><code>distro=ubuntu1804
arch=x86_64
</code></pre></div><p>登録方法1</p>
<div class="code-block-container"><pre><code>wget https://developer.download.nvidia.com/compute/cuda/repos/$distro/$arch/cuda-keyring_1.0-1_all.deb

sudo dpkg -i cuda-keyring_1.0-1_all.deb
</code></pre></div><p>登録方法2 (非推奨)</p>
<div class="code-block-container"><pre><code>sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/$distro/$arch/3bf863cc.pub
</code></pre></div>

<p>ご指摘ありがとうございます．<br>
この点についてはgistの方では更新していたのですが，こちらの記事の内容は更新していませんでした．<br>
後ほど反映させていただきます．</p>


Discussion

リポジトリキーの更新がありました(Apr 28, 2022)。

インストール方法 (distro=ubuntu1804, arch=x86_64の場合)