From 965e2b095d7fe63496ff932a0eceb3f1afc029db Mon Sep 17 00:00:00 2001 From: Changming Sun Date: Wed, 16 Sep 2020 09:01:31 -0700 Subject: [PATCH] Update MCR CUDA docker image to 10.2 (#5181) --- dockerfiles/Dockerfile.cuda | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dockerfiles/Dockerfile.cuda b/dockerfiles/Dockerfile.cuda index b17bae31cd..0d8f076c08 100644 --- a/dockerfiles/Dockerfile.cuda +++ b/dockerfiles/Dockerfile.cuda @@ -4,8 +4,8 @@ # -------------------------------------------------------------- # Dockerfile to run ONNXRuntime with CUDA, CUDNN integration -# nVidia cuda 10.1 Base Image -FROM nvidia/cuda:10.1-cudnn7-devel +# nVidia cuda 10.2 Base Image +FROM nvidia/cuda:10.2-cudnn8-devel MAINTAINER Vinitra Swamy "viswamy@microsoft.com" ARG ONNXRUNTIME_REPO=https://github.com/Microsoft/onnxruntime