Nvidia Docker安装说明(含WSL2)

• 订阅patreon即可看到更多文章 https://www.patreon.com/wade3c
• Youtube频道 https://www.youtube.com/c/增广建文

Linux下如果还没装过docker的话可以参考

https://docs.docker.com/engine/install/ubuntu/

curl -fsSL https://get.docker.com -o get-docker.sh

sudo sh get-docker.sh

WSL2

• WSL2在多次更新後也适用和Linux一样的安装方式了

• 21H2後也不用去安装特殊版的Nvidia driver了

• docker desktop在3.1版後内建支援WSL2上的nvidia-docker

WSL内的Docker也可以用Linux的装法，但有时候可能会遇到小问题

Linux

要先把nvidia的package server给加进apt的source list

distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
      && curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
      && curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | \
            sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
            sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list

然後就可以直接去装nvidia-docker套件了

sudo apt-get update
sudo apt-get install -y nvidia-docker2

装好後记得要去把旧有的docker服务给关掉

sudo service docker restart

用Ubuntu装的话就不用去设定/etc/docker/daemon.json

"default-runtime":"nvidia","runtimes":{"nvidia":{"path":"nvidia-container-runtime","runtimeArgs":[]}}

Test

有多种image可以进行测试，挑一个自己喜欢的就行

最简单的就是去看nvidia-smi能不能顺利运作

sudo docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi

Reference

Installation Guide — NVIDIA Cloud Native Technologies documentation

WSL 2 GPU Support for Docker Desktop on NVIDIA GPUs - Docker