saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-27 20:02:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update MCR CUDA docker image to 10.2 (#5181)

Browse source
This commit is contained in:
Changming Sun 2020-09-16 09:01:31 -07:00 committed by edgchen1
parent ece9a7c1fc
commit 5b5bcba9e3
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
dockerfiles/Dockerfile.cuda
View file

@ -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