0
0

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

More than 1 year has passed since last update.

Ubuntu20.04でのNVIDIA Docker環境構築

Last updated at Posted at 2022-01-19

Ubuntu20.04 + gpu docker環境を構築したので流れをメモしておきます。最後がポイントです。

環境

  • Ubuntu 20.04 (サーバーインストール)
  • NVIDIA GPU

手順

GPUドライバーインストール

昔はいろいろやった気がしたのですが、レポジトリーを追加して一気にいけます。

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt install ubuntu-drivers-common
sudo ubuntu-drivers autoinstall

動作確認は再起動後nvidia-smiで確認。ずらずら出てきます。

CUIログイン

サーバー用としてインストールしてもインストールするとするとGUI環境のログインが表示されてしまうので、CUIにしたい場合はログイン方法を変更します。

# 確認
$ sudo systemctl get-default
graphical.target

# 設定変更
$ sudo systemctl set-default multi-user.target

# 確認
$ sudo systemctl get-default
multi-user.target

Dockerインストール

通常の手順通りでインストールします。

  • sudo無しでdockerが動作するように変更
$ sudo usermod -aG docker $USER
  • 動作確認
# normal docker
docker run hello-world
  • GPU環境動作確認
docker run —gpus all nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04 nvidia-smi

nvidia-smiの画面が出ずに失敗します。dockerでGPUが使えるようになっていないようです。

docker: Error response from daemon: could not select device driver “” with capabilities: [[gpu]]の解消方法

curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | \
  sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list
sudo apt-get update
sudo apt-get install nvidia-container-runtime

service docker restartで再起動

  • 再度dockerのGPU動作を確認します。
docker run —gpus all nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04 nvidia-smi

OK!!

参考:

ごにょごにょ

ハード側の問題もあり結局数回インストールしなおす羽目になったので、メモしてありました。また気が向いたらクリーンインストールするかもしれないので、メモを公開しておきます。

0
0
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
0
0

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?