switch CUDA image to nvidia base
This commit is contained in:
parent
e704db50b1
commit
e9553bc610
|
@ -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
|
||||
|
||||
|
|
Loading…
Reference in New Issue