build-gpu

Dockerfile.gpu

 FROM nvidia/cuda:11.7.0-base-ubuntu22.04
 
-RUN DEBIAN_FRONTEND=noninteractive apt-get update && apt-get install -y gcc python3-dev python3-venv python3-pip
+RUN apt-get update -y && apt-get install -y gcc python3-dev python3-venv python3-pip
 
 COPY requirements.txt requirements.txt
 RUN python3 -m pip install -r requirements.txt && rm requirements.txt