saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix docker gpu test for csharp package cuda 9.1 and 10 (#432)

Browse source 
* Fix docker gpu test for csharp package cuda 9.1 and 10

* correct docker file name
This commit is contained in:
jignparm 2019-02-08 14:02:38 -08:00 committed by GitHub
parent 7b37dc6105
commit 5d00b8b375
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
csharp/test/Microsoft.ML.OnnxRuntime.EndToEndTests/runtest-docker-gpu.sh
View file

@ -4,20 +4,26 @@
# build docker image for CPU
#TODO: Get this working, not tested yet
set -x
SOURCE_ROOT=$1
BUILD_DIR=$2
NUGET_REPO_DIRNAME=$3 # path relative to BUILD_DIR
IMAGE="ubuntu16.04-cuda10.0-cudnn7.3"
#CUDA_VER=cuda10.0-cudnn7.3, cuda9.1-cudnn7.1
CUDA_VER=${4:-cuda9.1-cudnn7.1}
IMAGE="ubuntu16.04-$CUDA_VER"
PYTHON_VER=3.5
OldDir=$(pwd)
cd $SOURCE_ROOT/tools/ci_build/github/linux/docker
docker build -t "onnxruntime-$IMAGE" --build-arg OS_VERSION=16.04 --build-arg PYTHON_VERSION=${PYTHON_VER} -f Dockerfile.ubuntu_gpu .
cd $SOURCE_ROOT/tools/ci_build/github/linux/docker
DOCKER_FILE=Dockerfile.ubuntu_gpu_cuda9
if [ $CUDA_VER = "cuda10.0-cudnn7.3" ]; then
DOCKER_FILE=Dockerfile.ubuntu_gpu_cuda
fi
docker build -t "onnxruntime-$IMAGE" --build-arg OS_VERSION=16.04 --build-arg PYTHON_VERSION=${PYTHON_VER} -f $DOCKER_FILE .
docker rm -f "onnxruntime-gpu-container" || true
@ -40,7 +46,5 @@ wait -n
EXIT_CODE=$?
set -e
exit $EXIT_CODE
cd $OldDir
exit $EXIT_CODE