前言

由于一些库的需要，重新装了一个Linux环境，为了避免各种包冲突导致环境错乱，研究了一下Docker中使用GPU的方法。

Step 1 安装 Nvidia 驱动

sudo mhwd -a pci nonfree 0300

之后，重启

Step 2 安装 Docker

yay -S docker
sudo usermod -aG docker [your-name]
sudo systemctl start docker
sudo systemctl enable docker

Step 3 安装 nvidia-docker

sudo pacman -S nvidia-container-runtime

ps: sudo pacman -S 可以替换为 pacin 当开启 zsh 的 archlinx 插件后

Step 4 测试

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

可以看到输出正常

参考

• ArchWiki