Skip to content

Commit

Permalink
Merge branch 'devel' into mrcnn_to_tf2
Browse files Browse the repository at this point in the history
  • Loading branch information
chadhat authored May 26, 2021
2 parents 491f3e2 + 84c70b5 commit 98af2f7
Show file tree
Hide file tree
Showing 4 changed files with 495 additions and 85 deletions.
5 changes: 4 additions & 1 deletion Dockerfile
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
#FROM nvidia/cuda:11.2.2-cudnn8-runtime-ubuntu20.04

#FROM nvidia/cuda:11.2.2-cudnn8-runtime-ubuntu18.04
FROM nvidia/cuda:11.0.3-cudnn8-runtime-ubuntu20.04

MAINTAINER tarun.chadha@id.ethz.com

RUN apt-get -y update && apt-get -y upgrade
Expand Down Expand Up @@ -43,7 +45,7 @@ RUN git clone https://github.com/damaggu/SIPEC.git

WORKDIR /home/user/SIPEC

RUN git checkout mrcnn_to_tf2
RUN git checkout devel

ENV VIRTUAL_ENV=/home/user/SIPEC/env

Expand All @@ -52,6 +54,7 @@ RUN python3.7 -m venv env
ENV PATH="$VIRTUAL_ENV/bin:$PATH"

RUN python -m pip install --upgrade pip && \

pip install -r requirements.txt

ENV PYTHONPATH="/home/user/SIPEC:${PYTHONPATH}"
Expand Down
Loading

0 comments on commit 98af2f7

Please sign in to comment.