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