要使docker容器能调用本地的gpu,有三种种方法。
一是,安装nvidia-docker,该方法已被官方舍弃,地址为https://github.com/NVIDIA/nvidia-docker#upgrading-with-nvidia-docker2-deprecated
二是,安装nvidia-container-toolkit,后添加—gpus参数来使用,地址为:
安装https://github.com/NVIDIA/nvidia-docker#ubuntu-16041804-debian-jessiestretchbuster
使用https://github.com/NVIDIA/nvidia-docker#usage
三是,安装nvidia-container-runtime,在首次运行时添加—runtime=nvidia参数,后续启动、结束都不需要再加。
安装使用流程为:1
2
3
4
5
6
7
8
9
10
11
12
13
14apt-get install nvidia-container-runtime
tee /etc/docker/daemon.json <<EOF
{
"runtimes": {
"nvidia": {
"path": "/usr/bin/nvidia-container-runtime",
"runtimeArgs": []
}
}
}
EOF
docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi #正常就可以了