前言
由于一些库的需要,重新装了一个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
可以看到输出正常