Dockerfile

FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-devel
RUN apt-key adv --keyserver keyserver.ubuntu.com --recv-keys A4B469963BF863CC
RUN apt-get update && \
    apt-get upgrade -y && \
    apt-get install -y git
RUN apt-get -y install libgl1-mesa-glx libglib2.0-0
RUN apt-get -y install vim byobu aria2

COPY . /usr/src/motorbike_cls
# RUN ls /usr/src/motorbike_cls
RUN cd /usr/src/motorbike_cls

WORKDIR /usr/src/motorbike_cls

RUN pip install --no-cache-dir --upgrade pip && \
    pip install -r /usr/src/motorbike_cls/requirements.txt

CMD ["test.py"]
ENTRYPOINT ["python3"]