19 lines
523 B
Docker
19 lines
523 B
Docker
FROM nvidia/cuda:12.2.0-runtime-ubuntu22.04 AS requirements
|
|
RUN apt-get update
|
|
RUN apt-get update -y
|
|
RUN apt install -y python3-pip
|
|
ENV TOKENIZERS_PARALLELISM=false
|
|
COPY requirements.txt /requirements.txt
|
|
RUN pip3 install -r /requirements.txt
|
|
RUN rm -rf /requirements.txt
|
|
|
|
FROM requirements AS test
|
|
COPY ./ /root/sia/
|
|
WORKDIR /root/sia/
|
|
RUN mkdir -p /root/model
|
|
CMD ["python3", "-m", "unittest", "discover", "-v", "-p", "*test.py", "-v"]
|
|
|
|
FROM requirements
|
|
COPY ./ /root/sia/
|
|
WORKDIR /root/sia/
|
|
CMD ["python3", "-m", "sia"] |