From a3fe2ca559e8c668b44220b41c4cdd2ada317e01 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Wed, 11 Jan 2023 17:21:46 -0600 Subject: [PATCH] feat(api): set up venv in CUDA container, add onnxruntime-gpu --- api/Containerfile.cuda.buster | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/api/Containerfile.cuda.buster b/api/Containerfile.cuda.buster index eb590baea..5254be5da 100644 --- a/api/Containerfile.cuda.buster +++ b/api/Containerfile.cuda.buster @@ -2,13 +2,17 @@ FROM docker.io/python:3.10-buster WORKDIR /onnx-web -# TODO: set up venv within the container +RUN python3 -m venv onnx_env -RUN pip3 install torch torchvision torchaudio +ENV PATH="/onnx-web/onnx_web/bin:$PATH" COPY requirements.txt /onnx-web/requirements.txt -RUN pip install -r requirements.txt +RUN pip3 install -r requirements.txt + +RUN pip3 install onnxruntime-gpu + +RUN pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu117 COPY onnx_web/ /onnx-web/onnx_web/