saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-01 23:30:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Switch to use nvidia-docker2 command format

Browse source
This commit is contained in:
Changming Sun 2019-07-10 10:48:42 -07:00
parent a7fcd60572
commit 20f6c84fd2
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tools/ci_build/github/linux/run_dockerbuild.sh
View file

@ -80,13 +80,13 @@ if [ -z "$NIGHTLY_BUILD" ]; then
fi
if [ $BUILD_DEVICE = "cpu" ] || [ $BUILD_DEVICE = "ngraph" ] || [ $BUILD_DEVICE = "openvino" ]; then
ONNX_DOCKER=docker
RUNTIME=
else
ONNX_DOCKER=nvidia-docker
RUNTIME="--runtime=nvidia"
fi
docker rm -f "onnxruntime-$BUILD_DEVICE" || true
$ONNX_DOCKER run -h $HOSTNAME \
docker run $RUNTIME -h $HOSTNAME \
--name "onnxruntime-$BUILD_DEVICE" \
--volume "$SOURCE_ROOT:/onnxruntime_src" \
--volume "$BUILD_DIR:/build" \