docker容器使用gpu的環境配置

在docker中使用gpu，必須在建立容器時開啟--gpus引數，並保證docker的版本在19.03以上。

宿主機中建立檔案

vi nvidia-container-runtime-script.sh

新增以下內容，更新nvidia container runtime安裝包

sudo curl -s -l  | \

sudo apt-key add -
distribution=$(. /etc/os-release;echo $id$version_id)
sudo curl -s -l | \
sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list
sudo apt-get update

執行檔案

sh nvidia-container-runtime-script.sh

安裝nvidia-container-runtime

apt-get install nvidia-container-runtime

重啟docker

systemctl restart docker

驗證下gpu選項

docker run --help | grep -i gpus

執行帶gpus引數的ubuntu容器

docker run -it --rm --gpus all ubuntu:18.04

docker容器使用GPU方法

為了使docker容器可以使用nvidia docker,可以有如下幾種辦法 1 無nvidia docker 在早期的時候，沒有nvidia docker，可以通過在容器內再部署一遍nvidia gpu驅動解決。同理，其他裝置如果想在容器裡使用，也可以採用在容器裡重新安裝一遍驅動解決。2 nvid...

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...