diff --git a/Dockerfile.cpu b/Dockerfile.cpu new file mode 100644 index 0000000..b437e1e --- /dev/null +++ b/Dockerfile.cpu @@ -0,0 +1,22 @@ +FROM python:3.8 + +RUN apt update -y + +RUN apt install libgl1 -y + +RUN pip3 install \ + torch==1.10.2+cpu \ + torchvision==0.11.3+cpu \ + torchaudio==0.10.2+cpu \ + -f https://download.pytorch.org/whl/cpu/torch_stable.html + +RUN pip3 install \ + opencv-python \ + sklearn \ + matplotlib \ + pandas \ + tqdm + +#COPY src /src + +ENTRYPOINT ["bash"] \ No newline at end of file