使用 nvidia-smi 查看驱动版本,并在官网下载对应的驱动,以便于在容器中安装和宿主机相同版本的 NVIDIA 驱动。
为容器添加所有 GPU:
lxc config device add gpu gpu
添加指定 GPU:
lxc config device add gpu0 gpu id=0
在容器中安装显卡驱动:
sudo sh /NVIDIA-Linux-x86_64-xxx.xx.run --no-kernel-module
因为在容器中显卡驱动不需要安装内核文件,所以后面加上 --no-kernel-module。
在容器中安装好显卡驱动重启后输入 nvidia-smi 检查驱动是否安装成功