15 lines
589 B
Bash
15 lines
589 B
Bash
IMAGE=embodiedgen:v1.10.1.dev
|
|
CONTAINER=EmbodiedGen-docker-dev-${USER}
|
|
GPU_ID=6 # 在这里指定 GPU ID
|
|
|
|
docker run -itd --shm-size="64g" --gpus "device=${GPU_ID}" --cap-add=SYS_PTRACE \
|
|
--security-opt seccomp=unconfined --privileged --net=host \
|
|
-e GPUS=${GPU_ID} \
|
|
-v /data1/liy/projects/embodiedgen_data/inputs:/EmbodiedGen/inputs \
|
|
-v /data1/liy/projects/embodiedgen_data/outputs:/EmbodiedGen/outputs \
|
|
-v /data1/liy/projects/embodiedgen_data/weights:/EmbodiedGen/weights \
|
|
-v /data1/liy/projects/embodiedgen_data/logs:/EmbodiedGen/logs \
|
|
--name ${CONTAINER} ${IMAGE}
|
|
|
|
|