mirror of https://github.com/skygpu/skynet.git
Improve wrapdocker call method, make OSError not stop worker
parent
8a5e32d452
commit
769332f610
|
@ -15,9 +15,6 @@ run apt-get update -qq && apt-get install -qqy \
|
|||
|
||||
run curl -sSL https://get.docker.com/ | sh
|
||||
|
||||
add ./wrapdocker /usr/local/bin/wrapdocker
|
||||
run chmod +x /usr/local/bin/wrapdocker
|
||||
|
||||
volume /var/lib/docker
|
||||
|
||||
env HF_HOME hf_home
|
||||
|
@ -33,4 +30,7 @@ run pip install -r requirements.cuda.txt
|
|||
run pip install -r requirements.txt
|
||||
run pip install -e .
|
||||
|
||||
cmd ["wrapdocker"]
|
||||
add ./wrapdocker /usr/local/bin/wrapdocker
|
||||
run chmod +x /usr/local/bin/wrapdocker
|
||||
|
||||
cmd ["wrapdocker", "bash"]
|
||||
|
|
|
@ -27,7 +27,8 @@ async def failable(fn: partial, ret_fail=None):
|
|||
|
||||
except (
|
||||
asks.errors.RequestTimeout,
|
||||
json.JSONDecodeError
|
||||
json.JSONDecodeError,
|
||||
OSError
|
||||
):
|
||||
return ret_fail
|
||||
|
||||
|
|
|
@ -108,6 +108,5 @@ else
|
|||
fi
|
||||
sleep 1
|
||||
done
|
||||
[[ $1 ]] && exec "$@"
|
||||
exec bash --login
|
||||
[[ $1 ]] && exec bash -c "$@"
|
||||
fi
|
||||
|
|
Loading…
Reference in New Issue