Force nvidia runtime
This commit is contained in:
@@ -183,6 +183,7 @@ log "Starting Tentacle worker..."
|
|||||||
docker run -d \
|
docker run -d \
|
||||||
--name "${CONTAINER_NAME}" \
|
--name "${CONTAINER_NAME}" \
|
||||||
--network host \
|
--network host \
|
||||||
|
--runtime=nvidia \
|
||||||
--gpus all \
|
--gpus all \
|
||||||
--restart unless-stopped \
|
--restart unless-stopped \
|
||||||
-e CONDUCTOR_ADDR="${ORCH_ADDR}" \
|
-e CONDUCTOR_ADDR="${ORCH_ADDR}" \
|
||||||
|
|||||||
Reference in New Issue
Block a user