diff --git a/Dockerfile.gpu b/Dockerfile.gpu index c5de252e6f50ecbac2942134221ecfff3341876c..6d43a3ebec7554d6f07a076b0cfb9a0806b664e1 100644 --- a/Dockerfile.gpu +++ b/Dockerfile.gpu @@ -3,6 +3,7 @@ FROM nvidia/cuda:11.7.0-base-ubuntu22.04 RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/3bf863cc.pub 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