Added CPU dockerfile
This commit is contained in:
parent
36a2242b73
commit
b2f8b7606a
1 changed files with 22 additions and 0 deletions
22
Dockerfile.cpu
Normal file
22
Dockerfile.cpu
Normal file
|
@ -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"]
|
Reference in a new issue