apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

switch CUDA image to nvidia base

This commit is contained in:
Sean Sube 2023-01-11 18:15:22 -06:00
parent e704db50b1
commit e9553bc610
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/Containerfile.cuda.buster
View File

@ -1,4 +1,7 @@
FROM docker.io/python:3.10-buster
FROM docker.io/nvidia/cuda:11.7.1-runtime-ubuntu20.04
RUN apt -y update \
&& apt -y install python3 python3-pip python3-venv
WORKDIR /onnx-web