docker容器使用GPU方法

2021-10-07 08:10:46 字數 1471 閱讀 1344

為了使docker容器可以使用nvidia-docker,可以有如下幾種辦法:

1、無nvidia-docker

在早期的時候,沒有nvidia-docker,可以通過在容器內再部署一遍nvidia gpu驅動解決。同理,其他裝置如果想在容器裡使用,也可以採用在容器裡重新安裝一遍驅動解決。

2、nvidia-docker1.0

nvidia-docker是英偉達公司專門用來為docker容器使用nvidia gpu而設計的,設計方案就是把宿主機的gpu驅動檔案對映到容器內部使用,可以通過tensorflow生成gpu驅動資料夾。

3、nvidia-docker2.0

nvidia-docker2.0對nvidia-docker1.0進行了很大的優化,不用再對映宿主機gpu驅動了,直接把宿主機的gpu執行時對映到容器即可。啟動方式示例:

nvidia-docker run -d -e nvidia_visible_devices=all --name nvidia_docker_test  nvidia/cuda:10.0-base /bin/sh -c "while true; do echo hello world; sleep 1; done"
4、安裝docker-19.03及以上版本

docker19.03及以上版本,已經內建了nvidia-docker,無需再單獨部署nvidia-docker了。安裝方式如下:

安裝docker:

yum install -y yum-utils

yum-config-manager --add-repo

yum-config-manager --enable docker-ce-nightly

yum-config-manager --enable docker-ce-test

yum install docker-ce docker-ce-cli containerd.io

systemctl start docker

安裝nvidia-container-toolkit

distribution=

$(. /etc/os-release;

echo $id$version_id)

curl -s -l |

sudo

tee /etc/yum.repos.d/nvidia-docker.repo

sudo yum install -y nvidia-container-toolkit

sudo systemctl restart docker

啟動容器:

docker run --gpus all nvidia/cuda:10.0-base /bin/sh -c "while true; do echo hello world; sleep 1; done"
進入容器並輸入nvidia-smi驗證。

docker容器使用gpu的環境配置

在docker中使用gpu,必須在建立容器時開啟 gpus引數,並保證docker的版本在19.03以上。宿主機中建立檔案 vi nvidia container runtime script.sh 新增以下內容,更新nvidia container runtime安裝包 sudo curl s l...

Docker 容器使用

可以通過命令docker command help更深入的了解指定的 docker 命令使用方法。如果我們本地沒有 ubuntu 映象,我們可以使用 docker pull 命令來載入 ubuntu 映象 docker pull ubuntu以下命令使用 ubuntu 映象啟動乙個容器,引數為以命令...

Docker 容器使用

docker version 版本 systemctl status docker.service 狀態 1.檢視所有的容器命令 docker ps a2.啟動乙個已停止的容器 docker start 容器 id 3.停止容器的命令 docker stop 容器 id 4.停止的容器可以通過 do...