docker および nvidia-docker のインストール (深層学習環境の構成)

ドッカーのインストール:

ステップ 1: 必要なシステム ツールをインストールする

sudo apt-get update
sudo apt-get -y install apt-transport-https ca-certificates curl software-properties-common

ステップ 2: GPG 証明書をインストールする

curl -fsSL http://mirrors.aliyun.com/docker-ce/linux/ubuntu/gpg | sudo apt-key add -

ステップ 3: ソフトウェアソース情報を書き込む

sudo add-apt-repository "deb [arch=amd64] http://mirrors.aliyun.com/docker-ce/linux/ubuntu $(lsb_release -cs) stable"

ステップ 4: Docker-CE を更新してインストールする

sudo apt-get -y update
sudo apt-get -y install docker-ce

ステップ 5: コンテナー サービスを再起動します。

sudo systemctl daemon-reload
sudo systemctl restart docker

NVIDIA ドッカーをインストールする

ステップ 1: 古いバージョンの nvidia-docker をアンインストールします (インストールされている場合はスキップできます)

sudo docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f

sudo apt-get purge -y nvidia-docker

sudo apt autoremove

設定 2: apt キーを追加する

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -

ステップ 3: システムバージョンを追加する

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)

ステップ 4: ソースリストを更新する

curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

sudo apt-get update

ステップ 5: nvidia-docker をインストールする

sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd

ステップ 6: コンテナーを再起動する

sudo systemctl restart docker

テストデモ

sudo docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi

おすすめ

転載: blog.csdn.net/weixin_39274106/article/details/127102718