a | b/Dockerfile | ||
---|---|---|---|
1 | FROM pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime |
||
2 | |||
3 | WORKDIR /workspace |
||
4 | |||
5 | RUN apt-get update |
||
6 | RUN DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC apt-get -y install tzdata |
||
7 | RUN apt-get install build-essential git python3-opencv -y |
||
8 | |||
9 | # Debug |
||
10 | # COPY . /workspace |
||
11 | # RUN pip install . |
||
12 | |||
13 | # Release |
||
14 | RUN git clone https://github.com/MICLab-Unicamp/medseg |
||
15 | RUN pip install medseg/. |
||
16 | |||
17 | RUN mv /workspace/medseg/predict.sh /workspace/predict.sh |