mirror of
https://github.com/mapbox/robosat.git
synced 2025-12-08 20:14:01 +00:00
26 lines
857 B
Docker
26 lines
857 B
Docker
FROM nvidia/cuda:10.1-cudnn7-runtime
|
|
|
|
WORKDIR /usr/src/app
|
|
|
|
ENV LANG="C.UTF-8" LC_ALL="C.UTF-8" PATH="/opt/venv/bin:$PATH" PIP_NO_CACHE_DIR="false"
|
|
|
|
RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
|
|
python3 python3-pip python3-venv libspatialindex-c4v5 libglib2.0-0 && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
COPY requirements.txt .
|
|
|
|
RUN python3 -m venv /opt/venv && \
|
|
python3 -m pip install pip==19.1.1 pip-tools==3.7.0 && \
|
|
python3 -m piptools sync
|
|
|
|
RUN python3 -m pip install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp36m-linux_x86_64.whl && \
|
|
python3 -m pip install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp36-cp36m-linux_x86_64.whl
|
|
|
|
RUN python3 -c "from torchvision.models import resnet50; resnet50(True)"
|
|
|
|
COPY . .
|
|
|
|
ENTRYPOINT ["/usr/src/app/rs"]
|
|
CMD ["-h"]
|