导语:docker-compose启动GPU容器报错RuntimeError: Found no NVIDIA driver on your system.
明明docker启动没有问题,但是compose启动会报这个错。在容器内执行nvidia-smi,发现cuda版本没有显示
发现是compose版本太低,并且没有添加对应的参数。
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: all
capabilities: [gpu]
更新compose版本并启动
curl -L "https://github.com/docker/compose/releases/download/1.29.2/docker-compose-$(uname -s)-$(uname -m)" -o /usr/bin/docker-compose
docker-compose up -d