diff --git a/install.sh b/install.sh index 204a200..7516826 100644 --- a/install.sh +++ b/install.sh @@ -183,6 +183,7 @@ log "Starting Tentacle worker..." docker run -d \ --name "${CONTAINER_NAME}" \ --network host \ + --runtime=nvidia \ --gpus all \ --restart unless-stopped \ -e CONDUCTOR_ADDR="${ORCH_ADDR}" \