@@ -26,19 +26,19 @@ VENV_PYTHON=${DRIVER_PATH}/venv/bin/python
2626
2727if [ ! -f ${VENV_ACTIVATE} ]; then
2828 # Create python virtual enviornment
29- python3.8 -m venv " ${DRIVER_PATH} /venv"
29+ python3 -m venv " ${DRIVER_PATH} /venv"
3030fi
3131
3232# NOTE version
3333# - https://github.com/onnx/onnx/blob/master/docs/Versioning.md
3434# - https://github.com/onnx/onnx-tensorflow/blob/master/Versioning.md
3535
36- VER_TENSORFLOW=2.12.1
37- VER_ONNX=1.16 .0
38- VER_ONNXRUNTIME=1.18.0
39- VER_ONNX_TF=1.10.0
36+ VER_TENSORFLOW=2.19.0 # 2025-03-12
37+ VER_ONNX=1.17 .0
38+ VER_ONNXRUNTIME=1.20.1
39+ # VER_ONNX_TF=1.10.0
4040VER_PYDOT=1.4.2
41- VER_TORCH=" 2.1.2 +cpu"
41+ VER_TORCH=" 2.6.0 +cpu"
4242
4343# Install tensorflow
4444
6666fi
6767${VENV_PYTHON} -m pip ${PIP_OPTIONS} install Pillow
6868# Fix version to that of TF release date
69- ${VENV_PYTHON} -m pip ${PIP_OPTIONS} install tensorflow_probability==0.20.1
70- ${VENV_PYTHON} -m pip ${PIP_OPTIONS} install tensorflow_addons==0.20 .0
69+ # ${VENV_PYTHON} -m pip ${PIP_OPTIONS} install tensorflow_probability==0.25.0
70+ # ${VENV_PYTHON} -m pip ${PIP_OPTIONS} install tensorflow_addons==0.23 .0
7171
7272# Install PyTorch and ONNX related
7373# NOTE set ONE_PREPVENV_TORCH_SOURCE to override options for source URL.
@@ -84,14 +84,14 @@ ${VENV_PYTHON} -m pip ${PIP_OPTIONS} install onnx==${VER_ONNX}
8484${VENV_PYTHON} -m pip ${PIP_OPTIONS} install onnxruntime==${VER_ONNXRUNTIME}
8585
8686# Provide install of custom onnx-tf
87- if [ -n " ${EXT_ONNX_TF_WHL} " ]; then
88- ${VENV_PYTHON} -m pip ${PIP_OPTIONS} install ${EXT_ONNX_TF_WHL}
89- else
90- ${VENV_PYTHON} -m pip ${PIP_OPTIONS} install onnx-tf==${VER_ONNX_TF}
91- fi
87+ # if [ -n "${EXT_ONNX_TF_WHL}" ]; then
88+ # ${VENV_PYTHON} -m pip ${PIP_OPTIONS} install ${EXT_ONNX_TF_WHL}
89+ # else
90+ # ${VENV_PYTHON} -m pip ${PIP_OPTIONS} install onnx-tf==${VER_ONNX_TF}
91+ # fi
9292
9393# Fix version to that of TF release date
94- ${VENV_PYTHON} -m pip ${PIP_OPTIONS} install --upgrade protobuf==4.23.3
94+ ${VENV_PYTHON} -m pip ${PIP_OPTIONS} install --upgrade protobuf==5.29.4
9595
9696# Install pydot for visq
9797${VENV_PYTHON} -m pip ${PIP_OPTIONS} install pydot==${VER_PYDOT}
0 commit comments