Top > HPCTテクニカルセンター > NVIDIA Docker2のインストール方法

NVIDIA Docker2 のインストール方法

弊社で取り扱っているHPC-ProServerに対してnvidia-docker2をインストールする方法です。NVIDIAのドライバ及びCUDAはインストールされていること前提とします。また、NVIDIA GPU Cloud (NGC)へアカウント登録をし、API Keyの取得が必要となります。NGCへのアカウント登録については、以下のURLから可能です。

https://ngc.nvidia.com

​

(1) nvidia-docker2 レポジトリのインストール

​

rootにて以下のコマンドを実行します。


# distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
# curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | tee /etc/yum.repos.d/nvidia-docker.repo
 

インストール可能なバージョンは、以下のコマンドで確認できます。
 

# yum search --showduplicates nvidia-docker2


2018年9月6日現在、最新は、nvidia-docker2-2.0.3-1.docker18.06.1.ce.noarch でした。

​

(2) docker-ce レポジトリのインストール


rootにて以下のコマンドを実行します。


# yum-config-manager --add-repo https://download.docker.com/linux/centos/docker-ce.repo


インストール可能なdocker-ceのバージョンは、以下のコマンドで確認できます。


# yum search  --showduplicates docker-ce

​

ここでは、最新の18.06.1をインストールします。


# yum -y install docker-ce-18.06.1.ce-3.el7.x86_64

​

dockerの自動起動とサービスのスタート


# systemctl enable docker
# systemctl start docker

​

(3) NVIDIA Docker2のインストール


# yum install -y nvidia-docker2
# pkill -SIGHUP dockerd

​

(4) docker login


NGCで取得したAPI Keyを利用して、以下のコマンドでログインします。


# docker login nvcr.io
Username: $oauthtoken
Password: <Your Key>

​

(5) docker image の pull


NGCのメニューから、たとえばnvidia/tensorflowなどを見ると、pullするためのコマンドが記載されていますので、そのコマンドを実行します。