Skip to content

Commit

Permalink
ai readi image
Browse files Browse the repository at this point in the history
  • Loading branch information
quinnwai committed Apr 16, 2024
1 parent cd22d8e commit 47e97ce
Show file tree
Hide file tree
Showing 3 changed files with 77 additions and 0 deletions.
36 changes: 36 additions & 0 deletions ai-readi/Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,36 @@
# use base image
FROM quay.io/ohsu-comp-bio/bridge2ai-jupyter:restricted-gpu-root

# return to non-root user
USER $NB_UID
WORKDIR /home/$NB_USER

# create image from requirements.txt
RUN conda create --name ai-readi
SHELL ["conda", "run", "-n", "ai-readi", "/bin/bash", "-c"]
COPY --chown=$NB_USER:$NB_GID requirements.txt .
RUN pip install -r requirements.txt && \
rm requirements.txt

# setup retfound
RUN conda create -n retfound python=3.7.5 -y
SHELL ["conda", "run", "-n", "retfound", "/bin/bash", "-c"]
RUN git clone https://github.com/rmaphoh/RETFound_MAE/ && \
cd RETFound_MAE && \
pip install -r requirement.txt && \
cd $HOME

# setup automorph
RUN conda create -n automorph python=3.6 -y
SHELL ["conda", "run", "-n", "automorph", "/bin/bash", "-c"]
RUN git clone https://github.com/rmaphoh/AutoMorph.git && \
cd AutoMorph && \
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.1 -c pytorch -y && \
pip install --ignore-installed certifi && \
pip install -r requirement.txt && \
pip install efficientnet_pytorch

# allow use of conda envs
RUN conda init bash && exec bash

# USER root
3 changes: 3 additions & 0 deletions ai-readi/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
# AI-READi

Docker file for AI-READi. Builds off `restricted-gpu-root` image, reverting use back to non-root privileges. Creates environment with requirements.txt, a retfound environment, and a Automorph envrionment.
38 changes: 38 additions & 0 deletions ai-readi/requirements.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,38 @@
appnope==0.1.4
asttokens==2.4.1
comm==0.2.2
contourpy==1.2.1
cycler==0.12.1
debugpy==1.8.1
decorator==5.1.1
executing==2.0.1
fonttools==4.50.0
ipykernel==6.29.4
ipython==8.23.0
jedi==0.19.1
jupyter_client==8.6.1
jupyter_core==5.7.2
kiwisolver==1.4.5
matplotlib==3.8.3
matplotlib-inline==0.1.6
nest-asyncio==1.6.0
numpy==1.26.4
packaging==24.0
parso==0.8.3
pexpect==4.9.0
pillow==10.3.0
platformdirs==4.2.0
prompt-toolkit==3.0.43
psutil==5.9.8
ptyprocess==0.7.0
pure-eval==0.2.2
pydicom==2.4.4
Pygments==2.17.2
pyparsing==3.1.2
python-dateutil==2.9.0.post0
pyzmq==25.1.2
six==1.16.0
stack-data==0.6.3
tornado==6.4
traitlets==5.14.2
wcwidth==0.2.13

0 comments on commit 47e97ce

Please sign in to comment.