52 lines
1.5 KiB
Text
52 lines
1.5 KiB
Text
|
# FROM nvcr.io/nvidia/clara-train-sdk:v4.1
|
||
|
|
# FROM nvcr.io/nvidia/pytorch:22.08-py3
|
||
|
|
# FROM nvcr.io/nvidia/pytorch:24.01-py3
|
||
|
|
FROM nvcr.io/nvidia/tensorflow:24.03-tf2-py3
|
||
|
|
|
||
|
|
# apt
|
||
|
|
RUN apt-get update -y \
|
||
|
|
&& apt-get install libxrender1 openssh-server pigz sudo -y
|
||
|
|
|
||
|
|
# pip
|
||
|
|
|
||
|
|
# RUN pip install antspyx itk-elastix nipype nnunet nnunetv2 simpleitk==2.3.1 tensorflow[and-cuda] #too large .... install first
|
||
|
|
|
||
|
|
RUN pip install "torch<2.6" torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
|
||
|
|
RUN pip install antspyx itk-elastix nipype nnunet nnunetv2 simpleitk==2.3.1 #too large .... install first
|
||
|
|
|
||
|
|
# WORKDIR /workspace
|
||
|
|
WORKDIR /123
|
||
|
|
COPY requirements.txt /123
|
||
|
|
RUN pip install -r requirements.txt
|
||
|
|
|
||
|
|
RUN wget https://github.com/rordenlab/dcm2niix/releases/download/v1.0.20240202/dcm2niix_lnx.zip
|
||
|
|
RUN unzip dcm2niix_lnx.zip
|
||
|
|
RUN cp dcm2niix /usr/bin
|
||
|
|
|
||
|
|
# nnUNet
|
||
|
|
|
||
|
|
ENV nnUNet_raw_data_base="/workspace/nnUNet_raw_data_base"
|
||
|
|
ENV nnUNet_preprocessed="/workspace/nnUNet_preprocessed"
|
||
|
|
ENV RESULTS_FOLDER="/workspace/nnUNet_trained_models"
|
||
|
|
|
||
|
|
ENV nnUNet_raw="/workspace/v2/nnUNet_raw"
|
||
|
|
ENV nnUNet_preprocessed="/workspace/v2/nnUNet_preprocessed"
|
||
|
|
ENV nnUNet_results="/workspace/v2/nnUNet_results"
|
||
|
|
|
||
|
|
# SSH server
|
||
|
|
#RUN echo 'root:password' | chpasswd
|
||
|
|
#RUN echo "PasswordAuthentication yes" >> /etc/ssh/sshd_config
|
||
|
|
#RUN echo "PermitRootLogin yes" >> /etc/ssh/sshd_config
|
||
|
|
#ENTRYPOINT service ssh restart && env >> /etc/environment && bash
|
||
|
|
#EXPOSE 22
|
||
|
|
|
||
|
|
# Masonite
|
||
|
|
EXPOSE 8000
|
||
|
|
|
||
|
|
# jupyter
|
||
|
|
ENTRYPOINT jupyter-lab
|
||
|
|
EXPOSE 8888
|
||
|
|
|
||
|
|
# pynetdicom
|
||
|
|
EXPOSE 11120
|