Commit 39b097c7 authored by Ted Themistokleous's avatar Ted Themistokleous
Browse files

Move parity checks dependancies to dockerfile instead of script.

parent 75f92445
......@@ -125,3 +125,8 @@ ENV LD_LIBRARY_PATH=$PREFIX/lib
ENV UBSAN_OPTIONS=print_stacktrace=1
ENV ASAN_OPTIONS=detect_stack_use_after_return=1:check_initialization_order=1:strict_init_order=1
RUN ln -s /opt/rocm/llvm/bin/llvm-symbolizer /usr/bin/llvm-symbolizer
#install dependancies used for parity checks
RUN pip3 install psutil==5.9.5 onnx==1.10.2 coloredlogs==15.0.1 packaging==23.1 transformers==4.29.2 sympy==1.12
RUN pip3 install -U numpy==1.21.6
......@@ -125,3 +125,7 @@ ENV UBSAN_OPTIONS=print_stacktrace=1
ENV ASAN_OPTIONS=detect_stack_use_after_return=1:check_initialization_order=1:strict_init_order=1
RUN ln -s /opt/rocm/llvm/bin/llvm-symbolizer /usr/bin/llvm-symbolizer
#install dependancies used for onnxrt parity checks
RUN pip3 install psutil==5.9.5 onnx==1.10.2 coloredlogs==15.0.1 packaging==23.1 transformers==4.29.2 sympy==1.12
RUN pip3 install -U numpy==1.21.6
......@@ -23,10 +23,6 @@
#####################################################################################
set -e
#install dependancies used for parity checks
pip3 install psutil onnx coloredlogs packaging transformers==4.29.2 sympy
pip3 install -U numpy==1.21.6
#Go to folder with parity checks
cd /workspace/onnxruntime/onnxruntime/test/python/transformers/
......
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