
这个是启动命令:
version: '3.8'
services:
mgoat-attacker:
image: harbor.sourcefind.cn:5443/dcu/admin/base/vllm:0.9.2-ubuntu22.04-dtk25.04.2-1226-das1.7-py3.10-20251226
container_name: mgoat-attacker
network_mode: host
privileged: true
shm_size: 30g
devices:
- /dev/kfd
- /dev/dri
group_add:
- video
cap_add:
- SYS_PTRACE
security_opt:
- seccomp=unconfined
volumes:
- /opt/hyhal:/opt/hyhal
- /home/deploy/ai-safety-deploy/attacker-model:/model/attacker-model
environment:
- HIP_VISIBLE_DEVICES=0,1
- NCCL_DEBUG=info
- NCCL_P2P_DISABLE=1
- HSA_FORCE_FINE_GRAIN_PCIE=1
- NCCL_MIN_NCHANNELS=16
- NCCL_MAX_NCHANNELS=16
- VLLM_USE_FLASH_ATTN_PA=0
```
command: >
bash -c "python -m vllm.entrypoints.openai.api_server
--model /model/attacker-model
--served-model-name 'attacker-model'
--dtype 'bfloat16'
--host '0.0.0.0'
--port 32312
--trust-remote-code
--gpu-memory-utilization 0.85
--tensor-parallel-size 2
--max-num-seqs 16
--max-model-len 45000"
```