From 7ad093caf7fe835a3a13ce34674c6f30a3103470 Mon Sep 17 00:00:00 2001 From: zcr Date: Tue, 17 Mar 2026 12:29:50 +0800 Subject: [PATCH] 1 --- docker-compose.yaml | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/docker-compose.yaml b/docker-compose.yaml index a443441..9a65815 100644 --- a/docker-compose.yaml +++ b/docker-compose.yaml @@ -20,10 +20,12 @@ services: reservations: devices: - driver: nvidia - capabilities: [ gpu, compute, video ] - device_ids: [ '1' ] + device_ids: [ '0' ] + capabilities: [ gpu, compute, utility ] # 移除 video(非必需,减少兼容问题) command: > bash -c " /opt/conda/envs/trellis/bin/python -c 'import torch; print(torch.__version__, torch.cuda.is_available())' && - /opt/conda/envs/trellis/bin/python server.py - " \ No newline at end of file + /opt/conda/envs/trellis/bin/python server.py + " +# /opt/conda/envs/trellis/bin/python server.py +# tail -f /dev/null