1- # Get base from a pytorch image
2- FROM pytorch/pytorch:1.6.0-cuda10.1-cudnn7-runtime
1+ # Get base image
2+ FROM python:3.8.19-slim-bullseye
33
4- # Set to install things in non-interactive mode
5- ENV DEBIAN_FRONTEND noninteractive
6-
7- # Install system wide softwares
4+ # Install system wide software
85RUN apt-get update \
96 && apt-get install -y \
7+ build-essential \
108 libgl1-mesa-glx \
119 libx11-xcb1 \
1210 git \
@@ -17,45 +15,26 @@ RUN apt-get update \
1715 curl \
1816 libsndfile-dev \
1917 libsndfile1 \
18+ libhdf5-dev \
19+ gpg \
20+ gpg-agent \
2021 && apt-get install -y libsm6 libxext6 \
2122 && apt-get install -y libxrender-dev \
2223 && apt-get clean all \
2324 && rm -r /var/lib/apt/lists/*
2425
25- RUN /opt/conda/bin/conda install --yes \
26- astropy \
27- matplotlib \
28- pandas \
29- scikit-learn \
30- scikit-image
31-
3226# Install necessary libraries for goturn
33- RUN pip install torch==1.9.1
34- RUN pip install torchvision==0.10.1
35- RUN pip install tensorflow==2.6.0
36- RUN pip install pytest
37- RUN pip install numba
38- RUN pip install scikit-learn==0.20
39- RUN pip install pytest-cov
40- RUN pip install gdown
27+ RUN pip install torch==1.9.1 torchvision==0.10.1 tensorflow==2.9 scikit-learn==0.24 pytest~=6.2.5 pytest-cov~=4.1.0
4128
4229RUN git clone https://github.com/nrupatunga/goturn-pytorch.git /tmp/goturn-pytorch
4330# RUN cd /tmp/goturn-pytorch && pip install -r requirements.txt
44- RUN pip install loguru==0.5.3
45- RUN pip install torchsummary==1.5.1
46- RUN pip install tqdm==4.62.3
47- RUN pip install pytorch_lightning==0.7.1
48- RUN pip install imutils==0.5.3
49- RUN pip install torch_lr_finder==0.2.1
50- RUN pip install numpy==1.20.3
51- RUN pip install opencv_python==4.3.0.36
52- RUN pip install Pillow==8.0.1
53- RUN pip install visdom==0.1.8.9
54-
55- RUN pip install numpy==1.20.3
31+ RUN pip install loguru==0.5.3 torchsummary==1.5.1 tqdm==4.62.3 pytorch_lightning==0.7.1 imutils==0.5.3 torch_lr_finder==0.2.1 numpy==1.21.6 Pillow==8.0.1 visdom==0.1.8.9 gdown==5.1.0
32+ RUN pip3 install opencv_python==4.9.0.80 --no-build-isolation
5633
5734ENV PYTHONPATH "${PYTHONPATH}:/tmp/goturn-pytorch/src"
5835ENV PYTHONPATH "${PYTHONPATH}:/tmp/goturn-pytorch/src/scripts"
5936
6037RUN mkdir /tmp/goturn-pytorch/src/goturn/models/checkpoints
6138RUN cd /tmp/goturn-pytorch/src/goturn/models/checkpoints && gdown https://drive.google.com/uc?id=1GouImhqpcoDtV_eLrD2wra-qr3vkAMY4
39+
40+ CMD ["/bin/bash" ]
0 commit comments