File size: 773 Bytes
fa626df | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | FROM nvcr.io/nvidia/pytorch:22.12-py3
# FROM python:3.8.10-slim
# alternatively use python image as a base image, if PyTorch with GPU drivers is not needed
#COPY requirements.txt /script/requirements.txt
#COPY predict.py /script/predict.py
#COPY evaluate.py /script/evaluate.py
#COPY start.sh /script/start.sh
#COPY ./FungiCLEF2023-ViT_base_patch16_224-100E.pth /script/FungiCLEF2023-ViT_base_patch16_224-100E.pth
COPY . /script/
# install python dependencies
ENV SCRIPT_DIR='/script'
WORKDIR $SCRIPT_DIR
RUN pip install --no-cache-dir --upgrade pip build && \
pip install --no-cache-dir --compile -r requirements.txt && \
mim install "mmpretrain==1.0.0rc7" && \
rm -rf /var/lib/apt/lists/* /var/cache/apt/* /tmp/* /var/tmp/*
# run script
CMD bash start.sh
|