前言

由于一些库的需要,重新装了一个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 当开启 zsharchlinx 插件后

Step 4 测试

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

image-20201002211629171

可以看到输出正常

参考

Last modification:October 5th, 2020 at 08:35 pm