docker gpu环境配置
在docker中使用gpu,必须在创建容器时打开--gpus
参数,并保证docker的版本在19.03以上。
-
宿主机中创建文件
vi nvidia-container-runtime-script.sh
-
添加以下内容,更新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
-
执行文件
sh nvidia-container-runtime-script.sh
-
安装
nvidia-container-runtime
apt-get install nvidia-container-runtime
-
重启docker
systemctl restart docker
-
验证下gpu选项
docker run --help | grep -i gpus
-
运行带
gpus
参数的Ubuntu容器docker run -it --rm --gpus all ubuntu:18.04