Unverified Commit 9f374227 authored by Liangfu Chen's avatar Liangfu Chen Committed by GitHub
Browse files

[Neuron][CI] update docker run command (#14829)


Signed-off-by: default avatarLiangfu Chen <liangfc@amazon.com>
parent dd344e03
...@@ -44,7 +44,7 @@ remove_docker_container() { ...@@ -44,7 +44,7 @@ remove_docker_container() {
trap remove_docker_container EXIT trap remove_docker_container EXIT
# Run the image # Run the image
docker run --rm -it --device=/dev/neuron0 --device=/dev/neuron1 --network host \ docker run --rm -it --device=/dev/neuron0 --network bridge \
-v "${HF_CACHE}:${HF_MOUNT}" \ -v "${HF_CACHE}:${HF_MOUNT}" \
-e "HF_HOME=${HF_MOUNT}" \ -e "HF_HOME=${HF_MOUNT}" \
-v "${NEURON_COMPILE_CACHE_URL}:${NEURON_COMPILE_CACHE_MOUNT}" \ -v "${NEURON_COMPILE_CACHE_URL}:${NEURON_COMPILE_CACHE_MOUNT}" \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment