Top > HPCTテクニカルセンター > NVIDIA Docker2のインストール方法
NVIDIA Docker2 のインストール方法
弊社で取り扱っているHPC-ProServerに対してnvidia-docker2をインストールする方法です。NVIDIAのドライバ及びCUDAはインストールされていること前提とします。また、NVIDIA GPU Cloud (NGC)へアカウント登録をし、API Keyの取得が必要となります。NGCへのアカウント登録については、以下のURLから可能です。
(1) nvidia-docker2 レポジトリのインストール
rootにて以下のコマンドを実行します。
# distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
# curl -s -L | 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
インストール可能な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するためのコマンドが記載されていますので、そのコマンドを実行します。