forked from hansbu/quip_lung_cancer_detection
-
Notifications
You must be signed in to change notification settings - Fork 2
/
Dockerfile
28 lines (21 loc) · 943 Bytes
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
FROM pytorch/pytorch:1.0.1-cuda10.0-cudnn7-devel
MAINTAINER Tahsin Kurc
RUN apt-get -y update && \
apt-get install --yes python3-openslide wget zip libgl1-mesa-glx libgl1-mesa-dev && \
pip install --upgrade pip==21.0.1 && \
pip3 install setuptools==45 && \
pip3 install cython && \
conda install --yes pytorch=0.4.1 cuda90 -c pytorch && \
pip3 install scikit-learn && \
pip3 install "Pillow<7" pymongo pandas && \
pip3 install torchvision==0.2.1 && \
pip3 install opencv-python && \
pip3 install openslide-python
ENV BASE_DIR="/quip_app/quip_lung_cancer_detection"
ENV PATH="./":$PATH
ENV MODEL_VER="v1.0"
ENV MODEL_URL="https://github.com/SBU-BMI/quip_lung_cancer_detection/blob/master/models_cnn/train_lung_john_6classes_netDepth-34_APS-350_randomSeed-2954321_numBenign-80000_0131_1818_bestF1_0.8273143068611924_5.t7"
COPY . ${BASE_DIR}/.
RUN chmod 0755 ${BASE_DIR}/scripts/*
WORKDIR ${BASE_DIR}/scripts
CMD ["/bin/bash"]