ubuntu20.04でgpuをつかうdockerイメージを走らせるまで

1850 ワード
nvidia-docker2 NVIDIA Docker NVIDIA テキストリンク

はじめに

ubuntu 20.04 desktop + nvidia gpu + docker で環境構築したときのメモ

やることは以下の３つだ。

  • nvidiaドライバーのインストール
  • dockerのインストール
  • nvidia container toolkitのインストール

このリンク役に立つ。

nvidiaドライバーのインストール

初期状態だとnouveauというドライバーが入っているので、nvidiaのドライバをインストールする。

  • ソフトウェアとアップデートを起動
  • 追加のドライバー　タブを選択
  • nvidia...を使用します。をチェック
  • mokがなんたらかんたらでパスワードを作れと言われるのでパスワード作成
  • 再起動
  • 再起動するとなんか選択肢が現れるので「Enroll MOK」を選択
  • さっき作ったパスワードを入力

これで完成。MOKのところがよくわからんかったが、以下の記事を読んで解決した。

dockerのインストール

ここに従ってやれ。

nvidia container toolkitのインストール

ここに従ってやれ。

実行例

sudo docker run -it --gpus all --name study-ai -v ~/notebooks:/tf/notebooks -p 8888:8888 tensorflow/tensorflow:latest-jupyter

終わり。