前置条件

• Nvidia显卡
• 安装nvidia-container-runtime

安装nvidia-container-runtime

1. 添加Nvidia官方apt软件源（必须）

   创建脚本文件nvidia-container-runtime-script.sh

   sudo touch nvidia-container-runtime-script.sh
   

   脚本内添加以下内容

   # nvidia-container-runtime-script.sh
   
   sudo curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | \
     sudo apt-key add -
   distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
   sudo curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.list | \
     sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list
   sudo apt-get update
   

   运行脚本

   sudo sh nvidia-container-runtime-script.sh
   

2. apt安装

   sudo apt-get install nvidia-container-runtime
   

   安装完成后验证一下，输入：

   which nvidia-container-runtime
   

   应当得到输出：

   /usr/bin/nvidia-container-runtime