saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-19 21:32:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Use path relative to script location to refer to symbolic_opset10.py from install_deps.sh. (#3975)

Browse source 
Update install_deps.sh to use relative path from script directory to symbolic_opset10.py. This allows install_deps.sh to be called from different working directories.
This commit is contained in:
edgchen1 2020-05-18 13:36:06 -07:00 committed by GitHub
parent 9d2d1eb6f6
commit 024b92a970
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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/docker/scripts/install_deps.sh
View file

@ -1,6 +1,7 @@
#!/bin/bash
set -e
set -e -x
SCRIPT_DIR="$( dirname "${BASH_SOURCE[0]}" )"
while getopts p:d:x: parameter_Option
do case "${parameter_Option}"
@ -114,8 +115,7 @@ elif [ $DEVICE_TYPE = "gpu" ]; then
# patch pytorch onnx export opset version 10 to export nll_loss
PATH_TO_SYMBOLIC10=$(${PYTHON_EXE} -c 'import torch; import os; print(os.path.join(os.path.dirname(torch.__file__), "onnx/"))')
echo "cp ../scripts/pyt_patch/symbolic_opset10.py ${PATH_TO_SYMBOLIC10}"
cp ../scripts/pyt_patch/symbolic_opset10.py ${PATH_TO_SYMBOLIC10}
cp "${SCRIPT_DIR}/pyt_patch/symbolic_opset10.py" "${PATH_TO_SYMBOLIC10}"
fi
if [[ $BUILD_EXTR_PAR = *--enable_training_python_frontend_e2e_tests* ]]; then
${PYTHON_EXE} -m pip install transformers