Nvidia docker 安裝與使用

2021-10-11 08:17:39 字數 1832 閱讀 2681

nvidia docker 是nvidia顯示卡在docker基礎上進行封裝得到的docker工具,需要電腦中安裝nvidia顯示卡驅動與docker,配置好nvidia docker後docker可以使用gpu。本文記錄nvidia docker的安裝與使用方法。

安裝過程
curl -s -l  | \

sudo apt-key add -

distribution=$(. /etc/os-release;echo $id$version_id)

curl -s -l | \

sudo tee /etc/apt/sources.list.d/nvidia-docker.list

sudo apt-get update

sudo apt-get install -y nvidia-docker2

sudo pkill -sighup dockerd

docker run --runtime=nvidia --rm nvidia/cuda:11.0-base nvidia-smi
此時會顯示出顯示卡資訊,說明nvidia docker成功建立並在內部正確執行了nvidia-smi命令。

$ sudo apt show nvidia-docker2

->

package: nvidia-docker2

version: 2.5.0-1

priority: optional

section: utils

maintainer: nvidia corporation

installed-size: 27.6 kb

depends: nvidia-container-runtime (

>= 3.4.0), docker-ce (

>= 18.06.0~ce~3-0~ubuntu)

| docker-ee (

>= 18.06.0~ce~3-0~ubuntu)

| docker.io (

>= 18.06.0)

breaks: nvidia-docker (

<< 2.0.0)

replaces: nvidia-docker (

<< 2.0.0)

homepage:

download-size: 5,840 b

apt-manual-installed: yes

apt-sources: packages

replaces nvidia-docker with a new implementation based on

nvidia-container-runtime

n: there are 50 additional records. please use the '-a' switch to see them.

出現類似資訊說明安裝成功

nvidia-docker run -it --name first_container nvidia/cuda:11.0-base /bin/bash
這樣便將測試映象建立出了可以隨時訪問修改的容器,在該容器基礎上儲存映象即可。

docker commit -m "nvidia docker image init" first_container my_image:1.0

nvidia docker學習筆記

通過runfile安裝的nvidia驅動和cuda使用下列命令解除安裝 解除安裝驅動 sudo usr bin nvidia uninstall 解除安裝cuda sudo usr local cuda x.y bin uninstall cuda x.y.pl確認nouveau是否禁用 lsmod...

docker,以及nvidia docker安裝

更新apt包索引 2.安裝包以允許通過https使用儲存庫 3.新增docker的官方gpg金鑰 4.使用以下命令設定穩定儲存庫。即使您還想從邊緣或測試儲存庫安裝構建,您始終需要穩定的儲存庫。要新增邊緣或測試儲存庫,請在以下命令中的單詞stable之後新增單詞edge或test 或兩者 sudo a...

nvidia docker部署使用

sudo vim lib systemd system docker.service 新增 h tcp execstart usr bin dockerd h tcp h fd containerd run containerd containerd.sock sudo systemctl daem...