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