Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion get_deps.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down
6 changes: 3 additions & 3 deletions opt/build/onnxruntime/Makefile
Original file line number Diff line number Diff line change
@@ -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
Expand Down
8 changes: 2 additions & 6 deletions opt/build/onnxruntime/dockerfile.tmpl
Original file line number Diff line number Diff line change
@@ -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 %}

Expand All @@ -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 %}
Expand Down