hep-posttraining / container.sh
ho22joshua's picture
Upload container.sh with huggingface_hub
0711658 verified
Raw
History Blame Contribute Delete
325 Bytes
SHM_SIZE="${SHM_SIZE:-16g}"
IPC_MODE="${IPC_MODE:-host}"
IPC_ARGS=(--ipc="$IPC_MODE")
if [ "$IPC_MODE" != "host" ]; then
IPC_ARGS+=(--shm-size="$SHM_SIZE")
fi
podman-hpc run --rm -it --gpu --entrypoint= \
"${IPC_ARGS[@]}" \
-v "$PWD":/workspace \
-w /workspace \
docker.io/verlai/verl:sgl059.latest \
/bin/bash