前提
想在wsl2(Ubuntu20.04)环境中使用docker以及nvidia-docker来编译一些程序,这里对安装过程进行记录。
注意:wsl2中的安装docker和物理机器上安装docker的过程并不相同。因为wsl2方式的ubuntu里面没有systemd,所以通过apt insta docker-ce
命令安装的docker无法启动。
安装
这里我推荐使用Docker Desktop for windows方式来安装docker。
-
下载Docker Desktop for windows并安装
-
在setting中,找到下面的选项打上对勾
然后重启。 -
在wsl2中使用
docker version
命令查看docker是否安装成功。 -
添加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
-
安装nvidia-docker2
sudo apt-get update sudo apt-get install -y nvidia-docker2
-
测试是否安装成功
docker run --gpus all nvcr.io/nvidia/k8s/cuda-sample:nbody nbody -gpu -benchmark
出现如下信息则代表安装成功。
参考链接
- https://zhuanlan.zhihu.com/p/558821366
- https://zhuanlan.zhihu.com/p/148511634