mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-17 21:10:43 +00:00
24 lines
816 B
Text
24 lines
816 B
Text
|
#-------------------------------------------------------------------------
|
||
|
|
# Copyright (c) Microsoft Corporation. All rights reserved.
|
||
|
|
# Licensed under the MIT License.
|
||
|
|
#--------------------------------------------------------------------------
|
||
|
|
# Official user quickstart nvidia-docker container for ONNX Runtime GPU
|
||
|
|
# Ubuntu 16.04, GPU version, CuDNN 7, CUDA 10, Python 3.
|
||
|
|
#--------------------------------------------------------------------------
|
||
|
|
|
||
|
|
FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04
|
||
|
|
MAINTAINER Vinitra Swamy "viswamy@microsoft.com"
|
||
|
|
|
||
|
|
RUN apt-get update && \
|
||
|
|
apt-get install -y sudo \
|
||
|
|
build-essential curl \
|
||
|
|
libcurl4-openssl-dev \
|
||
|
|
libssl-dev wget \
|
||
|
|
python 3.6 python3-pip \
|
||
|
|
python3-dev git
|
||
|
|
RUN pip3 install --upgrade pip
|
||
|
|
RUN pip3 install numpy onnx
|
||
|
|
|
||
|
|
RUN pip3 install onnxruntime-gpu
|
||
|
|
WORKDIR /code
|