|
试试加runtime,现在docker daemon里添加:
tee /etc/docker/daemon.json <<-'EOF'
{
"runtimes": {
"nvidia": {
"path": "/usr/bin/nvidia-container-runtime",
"runtimeArgs": []
}
}
}
EOF
然后在docker-compose里再加一条:
runtime: nvidia
加的位置参考下面这个compose:
version: '2.3'
services:
nvsmi:
image: ubuntu:16.04
runtime: nvidia
environment:
- NVIDIA_VISIBLE_DEVICES=all
command: nvidia-smi |
|