前提

想在wsl2(Ubuntu20.04)环境中使用docker以及nvidia-docker来编译一些程序，这里对安装过程进行记录。

注意：wsl2中的安装docker和物理机器上安装docker的过程并不相同。因为wsl2方式的ubuntu里面没有systemd，所以通过apt insta docker-ce命令安装的docker无法启动。

安装

这里我推荐使用Docker Desktop for windows方式来安装docker。

1. 下载Docker Desktop for windows并安装

2. 在setting中，找到下面的选项打上对勾
   
   然后重启。

3. 在wsl2中使用docker version命令查看docker是否安装成功。

4. 添加nvidia-docker2的源

   distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
   curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
   curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
   curl -s -L https://nvidia.github.io/libnvidia-container/experimental/$distribution/libnvidia-container-experimental.list | sudo tee /etc/apt/sources.list.d/libnvidia-container-experimental.list
   

5. 安装nvidia-docker2

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

6. 测试是否安装成功

   docker run --gpus all nvcr.io/nvidia/k8s/cuda-sample:nbody nbody -gpu -benchmark
   

出现如下信息则代表安装成功。

参考链接

1. https://zhuanlan.zhihu.com/p/558821366
2. https://zhuanlan.zhihu.com/p/148511634