docker容器使用gpu的环境配置

docker gpu环境配置

在docker中使用gpu，必须在创建容器时打开--gpus参数，并保证docker的版本在19.03以上。

1. 宿主机中创建文件

   vi nvidia-container-runtime-script.sh
   

2. 添加以下内容，更新NVIDIA Container Runtime安装包

   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
   

3. 执行文件

   sh nvidia-container-runtime-script.sh
   

4. 安装nvidia-container-runtime

   apt-get install nvidia-container-runtime
   

5. 重启docker

   systemctl restart docker
   

6. 验证下gpu选项

   docker run --help | grep -i gpus
   

7. 运行带gpus参数的Ubuntu容器

   docker run -it --rm --gpus all ubuntu:18.04