diff --git a/get_deps.sh b/get_deps.sh index 3877cee4e..4aa3f46e4 100755 --- a/get_deps.sh +++ b/get_deps.sh @@ -2,7 +2,7 @@ ###### SET VERSIONS ###### -ORT_VERSION="1.9.0" +ORT_VERSION="1.11.1" DLPACK_VERSION="v0.5_RAI" TF_VERSION="2.8.0" TFLITE_VERSION="2.0.0" diff --git a/opt/build/onnxruntime/Makefile b/opt/build/onnxruntime/Makefile index f8cd57b05..a1dcb3916 100755 --- a/opt/build/onnxruntime/Makefile +++ b/opt/build/onnxruntime/Makefile @@ -1,16 +1,16 @@ -REDIS_ONNX_VERSION?=1.9.0 +REDIS_ONNX_VERSION?=1.11.1 REDIS_ONNX_REPO?=https://github.com/microsoft/onnxruntime PRODUCT=onnxruntime VERSION=${REDIS_ONNX_VERSION} -REDIS_CUDA_VERSION=11.0-cudnn8 +REDIS_CUDA_VERSION=11.3.1-cudnn8 # anything prefixed with REDIS, and exported is available in the docker build export REDIS_ONNX_VERSION export REDIS_ONNX_REPO export REDIS_CUDA_VERSION -OSNICK=xenial +OSNICK=bionic ROOT=. READIES=${ROOT}/../../readies diff --git a/opt/build/onnxruntime/dockerfile.tmpl b/opt/build/onnxruntime/dockerfile.tmpl index 2fda1a4ed..f85419822 100755 --- a/opt/build/onnxruntime/dockerfile.tmpl +++ b/opt/build/onnxruntime/dockerfile.tmpl @@ -1,6 +1,4 @@ -{% if REDIS_OSNICK == "xenial" %} -{% set cuda_suffix_os = "ubuntu16.04" %} -{% elif REDIS_OSNICK == "bionic" %} +{% if REDIS_OSNICK == "bionic" %} {% set cuda_suffix_os = "ubuntu18.04" %} {% endif %} @@ -12,9 +10,7 @@ FROM nvcr.io/nvidia/deepstream-l4t:5.1-21.02-base {% if REDIS_GPU is defined %} FROM nvidia/cuda:{{REDIS_CUDA_VERSION}}-devel-{{cuda_suffix_os}} {% else %} -{% if REDIS_OSNICK == "xenial" %} -FROM ubuntu:xenial -{% elif REDIS_OSNICK == "bionic" %} +{% if REDIS_OSNICK == "bionic" %} FROM ubuntu:bionic {% endif %} {% endif %}