saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-03 03:58:54 +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 GitHub
parent 79e27d937a
commit 965e2b095d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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