diff --git a/docker-compose.yml b/docker-compose.yml index 4bff40a..2bd4323 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -15,5 +15,5 @@ services: devices: # 告诉 Docker 使用所有可用的 NVIDIA GPU - driver: nvidia - device_ids: [ '0' ] + device_ids: [ '1' ] capabilities: [ gpu ] \ No newline at end of file