我用来docker compose运行一个容器:
version: "3.9"
services:
app:
image: nvidia/cuda:11.0.3-base-ubuntu20.04
deploy:
resources:
reservations:
devices:
- capabilities: [ gpu ]
Run Code Online (Sandbox Code Playgroud)
容器可以受益于 GPU 的存在,但它并不严格需要 GPU。使用上面的docker-compose.yaml结果会出错
来自守护程序的错误响应:无法选择具有以下功能的设备驱动程序“”:[[gpu]]
在没有 GPU 的机器上使用时。是否可以指定“使用 GPU(如果可用),否则启动没有 GPU 的容器”?